API Reference

This section provides detailed documentation for all Nabla modules and functions, organized to match the internal structure of the library.

API Documentation

• Core Module
  • Array
  • Execution Context
  • Graph Execution
  • Module Overview
• Ops Module
  • Kernels
  • Binary
  • Unary
  • Creation
  • Linalg
  • Reduce
  • View
  • Special
  • Operation
  • Conv Utils
  • Module Overview
• Transforms Module
  • Grad
  • Jacfwd
  • Jacrev
  • Jit
  • Jvp
  • Vjp
  • Vmap
  • Utils
  • Module Overview
• Nn Module
  • Layers
  • Losses
  • Optim
  • Init
  • Architectures
  • Utils
  • Module Overview
• Utils Module
  • Docs
  • Formatting
  • Grad Utils
  • Max Interop
  • Shape Utils
  • Testing
  • Types
  • Module Overview

Quick Navigation

Core Components

• Core Module - Array operations and execution context

• Ops Module - Low-level operations and kernels

Function Transformations

• Transforms Module - grad, jit, vmap, and other transformations

Neural Networks

• Nn Module - Layers, losses, optimizers, and architectures

Utilities

• Utils Module - Helper functions and utilities