Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Index Multiprocessing package - tensorplay.multiprocessing Python Module Index Quantization Search tensorplay tensorplay._stax tensorplay.audio tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision

latest (dev)

latest (dev)2026-09-08 v1.0.02026-09-06

Copy

Developer Notes

Automatic Mixed Precision examples
Autograd mechanics
Broadcasting semantics
Frequently Asked Questions
Gradcheck mechanics
Modules
Numerical accuracy
Out Notes
Reproducibility
Serialization semantics

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue