#language-model #burn #bdh #machine-learning

bin+lib burn_dragon_hatchling

burn (baby) dragon hatchling inference and training

2 unstable releases

Uses new Rust 2024

0.2.0 Oct 7, 2025
0.1.0 Oct 6, 2025

#1846 in Machine learning

MIT/Apache

510KB
2.5K SLoC

burn_dragon_hatchling 🔥🐉🐣

test GitHub License crates.io

burn inference and training of (baby) dragon hatchling model

Alt text Alt text

features

  • training benchmarks and reporting
  • adaptive tool discovery
  • conditional (deep) gating
  • document-coherent dataloading and scale mixup
  • episodic memory
  • fused kernels
  • hierarchical, memory-aware recurrent state
  • mixture-of-expert routing
  • multi-modal architecture
  • neuromorphic backend
  • rl reasoning training
  • streaming, sparse synaptic backpropagation
  • temporal neuron dampening

training

  • cargo run --release -- --config ./config/dev.toml (defaults to the cuda backend)

benchmarks

  • cargo bench (executes both wgpu and cuda benchmarks)
  • open target/criterion/report/index.html

compatible burn versions

burn_dragon_hatchling burn
0.1 0.18

license

licensed under either of

at your option.

contribution

unless you explicitly state otherwise, any contribution intentionally submitted for inclusion in the work by you, as defined in the Apache-2.0 license, shall be dual licensed as above, without any additional terms or conditions.

Dependencies

~82–125MB
~2M SLoC