/

Tensor

lucid.Tensor
lucid.tensor
Tensor Operations
Tensor Utilities

Autograd

lucid.autograd
Autograd APIs
- lucid.autograd.backward
- lucid.autograd.grad

Linear Algebra

lucid.linalg
Linalg Operations

Random

lucid.random
RNG Functions

Einstein Operations

lucid.einops
Einops Functions

Neural Networks

lucid.nn
nn.Parameter
nn.Buffer
nn.Module
Module Hooks
Neural Functions
Weight Initializations
Modules
Fused Modules
Containers
Caches
Utilities

Optimization

lucid.optim
optim.Optimizer
optim.lr_scheduler
Optimizers
LR Schedulers

Data

lucid.data
data.Dataset
data.Subset
data.TensorDataset
data.ConcatDataset
data.DataLoader
Tokenizers
Utilities
- data.random_split

Datasets

lucid.datasets
Image Datasets
- MNIST
- FashionMNIST
- CIFAR10
- CIFAR100

Models

lucid.models
Base Mixins
- PreTrainedModelMixin
Vision Models
Generative Models
- VAE
  - VAEConfig
- DDPM
  - DDPMConfig
- NCSN
  - NCSNConfig
Text Models
Model Utilities

Weights

lucid.weights
Pre-Trained Weights

Transformation

lucid.transforms
transforms.Compose
transforms.ToTensor
Image Transforms

Compilation

JIT Compilation
lucid.compile

Visualization

lucid.visual
Mermaid Charts
- visual.build_tensor_mermaid_chart
- visual.build_module_mermaid_chart

Porting

lucid.save
lucid.load

Others

lucid.Numeric
lucid.no_grad
lucid.grad_enabled
lucid.count_flops
lucid.newaxis
lucid.register_model

Lucid /
Neural Network Utilities

Open in ChatGPT

Neural Network Utilities¶

nn.utils.grad_norm
nn.utils.get_total_norm
nn.utils.clip_grad_norm
nn.utils.clip_grad_value
nn.utils.apply_chunking_to_forward
RNN Utilities

nn.StaticKVCache

nn.utils.grad_norm

2026, ChanLumerico

Made with Sphinx and Shibuya theme.