/pax

A stateful pytree library for training neural networks.

Primary LanguagePythonMIT LicenseMIT

Watchers