poets-ai / elegyLinks
A High Level API for Deep Learning in JAX
☆475Updated 2 years ago
Alternatives and similar repositories for elegy
Users that are interested in elegy are comparing it to the libraries listed below
Sorting:
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆423Updated 7 months ago
- ☆772Updated last year
- ☆885Updated last month
- ☆599Updated last week
- Annotating tensor shapes using Python types☆158Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.