bullet is a domain-specific ML library, generally used for training NNUE-style networks for some of the strongest chess engines in the world.
- Follows SemVer with crates.io releases
- Contains the
DeviceandDeviceBuffertraits, which are used to define a backend - Network graph construction, execution and autodiff generic over backends
- Follows SemVer with crates.io releases
- New CUDA backend
- Not currently in a useable state
- Follows SemVer with crates.io releases
- Contains both the HIP and CUDA backends
- Does not follow any particular versioning
- API is sometimes subject to breaking changes
NetworkTrainertraits wrap the core graph API, providing a training loop with data loading performed asynchronously from device calculations- Contains
Trainer, which implementsNetworkTrainerfor value network training - Contains
TrainerBuilder, which streamlines the process of constructing aTrainerfor the most common network architectures - Lots of NNUE and chess engine specific tooling
- Input feature types
- Output buckets
- Data formats
- Does not follow any particular versioning
- Is a CLI program with various chess related utilities
- Data validation
- Converting between data file types
- Shuffling and interleaving data files
Before attempting to use, check out the docs. They contain all the main information about building bullet, managing training data and the network output format.
Most people simply clone the repo and edit one of the examples to their taste.
Alternatively, import the bullet_lib crate with
bullet = { git = "https://github.com/jw1912/bullet", package = "bullet_lib" }Specific API documentation is covered by Rust's docstrings.
- Please open an issue to file any bug reports/feature requests.
- Feel free to use the dedicated
#bulletchannel in the Engine Programming discord server if you run into any issues. - For general training discussion the Engine Programming non-
#bulletchannels are appropriate, or#engines-devin the Stockfish discord.