Skip to content

Ava 0.1: her first trained brains

Choose a tag to compare

@nochallenge nochallenge released this 24 Oct 23:57
· 6 commits to main since this release

Ava's first public release: a brain-like language model in pure Rust, with leaky integrate-and-fire spiking neurons trained by surrogate gradients, k-winners-take-all sparse codes, and recurrent membrane memory instead of attention.

Model Size Trained on Held-out bits per byte
Ava-S (release/ava-s) d 192, 6 layers, 3.0M weights 48M tokens, 8 CPU threads, 2 h 06 min 1.872
Ava-M (release/ava-m) d 384, 8 layers, 13.4M weights 24M tokens, one RTX 4070, 21 min 1.888

Highlights

  • A dependency-free CPU stack (tokenizer, token shards, autodiff, threaded kernels, AdamW), written from the Rust standard library up.
  • A CUDA trainer on candle with a fused LIF kernel and a hand-written backward: 3.2× faster than the unrolled graph, and a memory pool that stays flat over a run.
  • The pre-registered v32 comparison of the spiking design against a transformer at three sizes.
  • Model cards with exact commands, seeds, checkpoint hashes, unselected samples, and every training attempt disclosed.

She is a tiny research model. She will often produce nonsense; do not rely on anything she says. See the README for what she is, how to run her, and the roadmap.

Weights are also in the repository; the SHA-256 sums are in each release/*/SHA256SUMS.

Licence: PolyForm Noncommercial 1.0.0 (source-available).