Nabla ML
High-performance autodiff and distributed ML, inspired by JAX and powered by Mojo & MAX.
pip install --pre --extra-index-url https://whl.modular.com/nightly/simple/ modular nabla-ml
🚀 Get Started
Learn the basics of Nabla, from tensors and autodiff to SPMD sharding and Mojo integration.
📚 API Reference
Detailed documentation for Nabla’s core modules, operations, and transforms.
💡 Examples
Explore guided notebooks demonstrating training, pipeline parallelism, and fine-tuning.