Equinox

JAX neural network library with PyTorch-like module system.

Open SourceSelf HostedOffline Capable
0.0 (0)

About

Equinox by Patrick Kidger brings a PyTorch-like module system to JAX. Filtered transformations replace pytrees with custom classes. Elegant and composable. Used in diffrax and other libraries. Apache 2.0 license.

Reviews (0)

Leave a Review

No reviews yet. Be the first to review!

Details

Price
Free
Platform
Local/Desktop
Difficulty
Advanced (4/5)
License
Apache-2.0
Added
Apr 3, 2026

Similar Tools

Featured

Open-source machine learning framework by Meta with dynamic computation graphs.

Open SourceSelf HostedOffline
Intermediate
0.0 (0)
Featured

End-to-end open-source ML platform by Google for training and deployment.

Open SourceSelf HostedOffline
Intermediate
0.0 (0)

High-performance numerical computing library by Google with auto-differentiation.

Open SourceSelf HostedOffline
Advanced
0.0 (0)