eazygrad.no_grad

eazygrad.no_grad(func: Callable[[...], Any]) → Callable[[...], Any]

Decorator that disables gradient tracking inside a function.

Parameters:

func (callable) – Function to execute with gradient tracking disabled.

Returns:

callable – Wrapped function.