Skip to main content
Back to top
Ctrl
+
K
Getting Started
Examples
Example 1: Tensors and Operations
Example 2: Automatic Differentiation
Example 3: Graph Tracing — Under the Hood
Example 4a: MLP Training — PyTorch-Style (Imperative)
Example 4b: MLP Training (JAX-Style / Functional)
Example 5: Transforms and
@nb.compile
Example 6a: Transformer Training — PyTorch-Style (Imperative)
Example 6b: Transformer Training (JAX-Style / Functional)
Example 7: Pipeline Parallelism (GPipe)
Example 8: 2D Parallel Training (Pipeline + Data Parallelism)
Example 9: Pipeline Parallel Inference
Example 10: Compiled vs Eager vs JAX
Example 11: LoRA & QLoRA Fine-Tuning
Example 12: Custom Mojo Kernels
Example 13: CNN Training
API Reference
Core
Tensor
Defaults & Context
Autograd Runtime
PyTrees
Sharding
Types & Devices
Transforms
Differentiation
Vectorization
Compilation
Distributed Transforms
Operations
Creation
Binary
Convolution & Pooling
Unary & Activations
Reduction
Comparison & Logical
Control Flow
Multi-Output
View & Shape
Communication
Custom Op Utilities
Neural Networks (nabla.nn)
Modules
Functional API
Optim
Fine-Tuning
LoRA
QLoRA
Checkpointing
Testing (nabla.testing)
Utilities
Mojo Integration
Custom Kernels
Repository
Suggest edit
Open issue
.md
.pdf
Testing (nabla.testing)
Testing (nabla.testing)
#
Utilities for test assertions and eager realization in test code.
Utilities
