Skip to main content
Back to top
Ctrl
+
K
API Reference
Tensor
Function Transformations
Automatic Differentiation
Vectorization
Compilation
Neural Networks (nabla.nn)
Modules
Base Class
Linear Layers
Activation Modules
Containers
Functional
Functional Activations
Functional Losses
Functional Layers
Optimizers (nabla.optim)
Core
Algorithms
Low-Level Tensor Operations (nabla.ops)
Binary Operations
Creation Operations
Indexing Operations
Linear Algebra Operations
Reduction Operations
Special Operations
Unary Operations
View & Manipulation Operations
Utilities
Device Management
Testing
Tutorials
A Primer on Automatic Differentiation
Training an MLP (CPU and GPU)
Training a Transformer (CPU and GPU)
Program Transformations Uncovered
Repository
Suggest edit
Open issue
.md
.pdf
Utilities
Utilities
#
Helper functions for device management and testing.
Device Management
Testing
