NTT123/pax

A stateful pytree library for training neural networks.

3 best points · 0 comments · 1 Hacker News post · 22 GitHub stars · View on GitHub

Hacker News discussions

• Pax: a PyTorch-like pytree module on top of Jax (3 points, 0 comments, by xcodevn, 2021-09-04)