eazygrad

Top-level API

Tensor Object

Tensor Creation

tensor

Create a tensor from a Python scalar, list, or NumPy array.

from_numpy

Create a tensor that shares storage with an existing NumPy array.

randn

Return a tensor filled with samples from the standard normal distribution.

uniform

Return a tensor filled with samples from a uniform distribution.

ones

Return a tensor filled with ones.

zeros

Return a tensor filled with zeros.

Optimization

SGD

Stochastic gradient descent optimizer.

Adam

Adam optimizer.

AdamW

AdamW optimizer with decoupled weight decay.

Autograd Control

no_grad

Decorator that disables gradient tracking inside a function.

no_grad_ctx

Context manager that temporarily disables gradient tracking.