equinox
Neural Networks in JAX.
Installation
To install this package, run one of the following:
Usage Tracking
0.13.6
0.13.5
0.13.4
0.13.3
0.13.2
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 15, 2026 at 00:21
License
Apache-2.0
Supported Platforms
noarch
GitHub Repository
https://github.com/patrick-kidger/equinoxDocumentation
https://docs.kidger.site/equinox/