13 lines
474 B
Plaintext
13 lines
474 B
Plaintext
JAX library for neural networks and PyTree manipulations.
|
|
|
|
Equinox is a JAX library that provides everything you need that isn't already
|
|
in core JAX:
|
|
- neural networks (or more generally any model), with easy-to-use
|
|
PyTorch-like syntax;
|
|
- filtered APIs for transformations;
|
|
- useful PyTree manipulation routines;
|
|
- advanced features like runtime errors;
|
|
|
|
Equinox isn't a framework: everything you write in Equinox is compatible with
|
|
anything else in JAX or the ecosystem.
|