API Reference

Every module in the library, generated from the doc comments and method signatures in the source. Run node docs/gen.mjs to regenerate this page and the module pages.

Core

• Util

Core

• PRNG

Core

• Tensor

Core

• Matrix

NN

• Module

NN

• Inits

NN

• Linear

NN

• Activations

NN

• Norm

NN

• Dropout

NN

• Embedding

NN

• RoPE

NN

• Attention

NN

• FFN

NN

• Sequential

Models

• MLP

Models

• TransformerBlock

Models

• Transformer

Loss

• CrossEntropy

Loss

• Losses

Optim

• AdamW

Optim

• AdEMAMix

Optim

• Lion

Optim

• NAdamW

Optim

• CautiousAdamW

Optim

• ScheduleFreeAdamW

Optim

• SOAP

Optim

• Muon

Optim

• MuonAdamW

Train

• Scheduler

Train

• EMA

Train

• Serialize

Train

• Trainer

Data

• BPE