Skip to main content
Back to top
Ctrl
+
K
Tutorials
Understanding Nabla - Program Transformations (Part 1)
Value-and-Grads (CPU)
Value-and-Grads (GPU)
MLP Training (CPU)
MLP Training (GPU)
JAX vs. Nabla: Training an MLP (CPU)
JAX vs. Nabla: Training a Transformer (CPU)
API Reference
Core Module
Array
Execution Context
Graph Execution
Ops Module
Kernels
Binary
Unary
Creation
Linalg
Reduce
View
Special
Operation
Conv Utils
Transforms Module
Grad
Jacfwd
Jacrev
Jit
Jvp
Vjp
Vmap
Utils
Nn Module
Layers
Activations
Linear
Losses
Regression
Classification
Optim
Sgd
Adamw
Adam
Schedules
Init
Variance Scaling
Architectures
Mlp
Utils
Metrics
Regularization
Training
Utils Module
Docs
Formatting
Grad Utils
Max Interop
Shape Utils
Testing
Types
Repository
Suggest edit
Open issue
.md
.pdf
Kernels
Contents
Submodule Overview
Kernels
#
Custom compute kernels
Submodule Overview
#
Contents
Submodule Overview
