burn_bdh 🔥🐉🐣

burn inference and training of baby dragon hatchling model

features

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

training

cargo run --release -- --config ./config/dev.toml

benchmarks

• cargo bench
• open target/criterion/report/index.html

compatible burn versions

license

licensed under either of

• Apache License, Version 2.0 (http://www.apache.org/licenses/LICENSE-2.0)
• MIT license (http://opensource.org/licenses/MIT)

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.