equinox
Neural Networks in JAX.
Installation
To install this package, run one of the following:
Usage Tracking
0.13.5
0.13.4
0.13.3
0.13.2
0.13.1
5 / 8 versions selected
Downloads (Last 6 months): 0
Description
Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.
About
Summary
Neural Networks in JAX.
Last Updated
Mar 7, 2026 at 14:54
License
Apache-2.0
Total Downloads
105.9K
Version Downloads
19
Supported Platforms
noarch
GitHub Repository
https://github.com/patrick-kidger/equinoxDocumentation
https://docs.kidger.site/equinox/