zenkai.utils

Tensor and parameter utilities: type conversion, gradient context managers, and the memory sub-package for storing batches of samples.

Modules

zenkai.utils.memory