API Reference

• Core
  • Tensor
  • Defaults & Context
  • Autograd Runtime
  • PyTrees
  • Sharding
  • Types & Devices
• Transforms
  • Differentiation
  • Vectorization
  • Compilation
  • Distributed Transforms
• Operations
  • Creation
  • Binary
  • Unary & Activations
  • Reduction
  • Comparison & Logical
  • Control Flow
  • Multi-Output
  • View & Shape
  • Communication
• Neural Networks (nabla.nn)
  • Modules
  • Functional API
  • Optim
  • Fine-Tuning
• Testing (nabla.testing)
  • Utilities
• Mojo Integration
  • Custom Kernels