Core
Core Operations
Tensor creation, indexing, and manipulation
Math
Mathematical operations and functions
Linear Algebra
Matrix operations, decompositions, and solvers
Neural Networks
Neural Networks
Layers, modules, and network building blocks
Loss Functions
Loss functions for training
Functional API
Stateless functional operations
Training
Optimization
Optimizers and learning rate schedulers
Autograd
Automatic differentiation engine
Data
Datasets, dataloaders, and samplers
Distributed & Hardware
Distributed
Multi-GPU and multi-node training
Device
CUDA, MPS, and device management
Random
Random number generation and seeding
Signal & Serialization
Signal
Signal processing operations
Serialization
Save and load models and tensors
Profiling
Performance profiling tools
Compilation & Export
Optimization & Transforms
Quantization
Model quantization for inference
Func Transforms
Functional transformations (vmap, grad)
Sparse
Sparse tensor operations
Specialized
Nested
Nested tensor operations
Special
Special mathematical functions
Utils
Utility functions and helpers