AI infrastructure.
Purpose-built.

// Fashion-MNIST: 86.2% test accuracy, <1s
// RTX 5090 Laptop — 900K samples/s (wall) · 1.3M samples/s (GPU)
class OaMnistClassifier : public OaModule {
public:
  OaMnistClassifier() {
    Fc1_ = Register(OaMakeShared<OaLinear>(784, 128));
    Fc2_ = Register(OaMakeShared<OaLinear>(128, 10));
  }
  OaTensor Forward(const OaTensor& x) override {
    auto h = OaFnMatrix::Scale(x, 1.0f / 255.0f);
    h = OaFnMatrix::Relu(Fc1_->Forward(h));
    return Fc2_->Forward(h);
  }
private:
  OaSharedPtr<OaLinear> Fc1_, Fc2_;
};

int main() {
  auto rt = OaEngine::Create({.AppName = "Mnist"}).Unwrap();
  OaMnistClassifier model;
  OaAdamW opt(model.Parameters(), 0.001f);

  for (OaI32 step = 0; step < 4685; ++step) {
    auto logits = model.Forward(batchImages);
    auto loss   = OaFnMatrix::CrossEntropyLoss(logits, labels);
    OaFnGrad::Backward(loss);
    opt.Step();
    opt.ZeroGrad();
  }
  model.Save("mnist.oam");
}

// Char-level text model — Level 1 autograd
// RTX 5090 Laptop — 320K samples/s (wall) · 820K samples/s (GPU)
// Final loss 0.041 · 96.9% batch accuracy
class OaTextModel : public OaModule {
public:
  OaTextModel() {
    Embed_ = Register(OaMakeShared<OaEmbedding>(27, 16));
    Fc1_   = Register(OaMakeShared<OaLinear>(128, 64));
    Head_  = Register(OaMakeShared<OaLinear>(64, 27));
  }
  OaTensor Forward(const OaTensor& x) override {
    auto e = Embed_->Forward(x);   // [batch*8, 16]
    auto h = OaFnMatrix::Tanh(
               Fc1_->Forward(e)); // [batch, 64]
    return Head_->Forward(h);     // [batch, 27]
  }
private:
  OaSharedPtr<OaEmbedding> Embed_;
  OaSharedPtr<OaLinear>    Fc1_, Head_;
};

int main() {
  auto rt = OaEngine::Create({.AppName = "Text"}).Unwrap();
  OaTextModel model;
  OaAdamW opt(model.Parameters(), 0.01f);

  for (OaI32 step = 0; step < 300; ++step) {
    auto logits = model.Forward(input);
    auto loss   = OaFnMatrix::CrossEntropyLoss(logits, targets);
    OaFnGrad::Backward(loss);
    opt.Step();
    opt.ZeroGrad();
  }
  model.Save("text.oam");
}