A functional training loops library for JAX
☆88Feb 13, 2024Updated 2 years ago
Alternatives and similar repositories for ciclo
Users that are interested in ciclo are comparing it to the libraries listed below
Sorting:
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 2 years ago
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Tools for JAX☆50Feb 20, 2026Updated last week
- Neural Networks for JAX☆84Sep 24, 2024Updated last year
- ☆24Jul 31, 2023Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆46Jan 11, 2025Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated 11 months ago
- Stencil computations in JAX☆71Nov 15, 2025Updated 3 months ago
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- ☆13Jul 12, 2024Updated last year
- JAX Arrays for human consumption☆110Oct 23, 2025Updated 4 months ago
- Pytrees + dataclasses ❤️☆75Dec 19, 2025Updated 2 months ago
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- Named Tensors for Legible Deep Learning in JAX☆217Nov 8, 2025Updated 3 months ago
- A scalable Dreamer implementation in JAX☆10May 22, 2022Updated 3 years ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Jul 20, 2024Updated last year
- A simple library for scaling up JAX programs☆146Nov 4, 2025Updated 4 months ago
- Flow-matching algorithms in JAX☆115Aug 12, 2024Updated last year
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆549Updated this week
- ☆624Jan 23, 2026Updated last month
- JMP is a Mixed Precision library for JAX.☆211Jan 30, 2025Updated last year
- Pytorch-like dataloaders for JAX.☆101Dec 16, 2025Updated 2 months ago
- Graph neural networks in JAX.☆68Jun 18, 2024Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆546Feb 16, 2026Updated 2 weeks ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆273Sep 22, 2025Updated 5 months ago
- LoRA for arbitrary JAX models and functions☆145Feb 26, 2024Updated 2 years ago
- JAX Synergistic Memory Inspector☆184Jul 16, 2024Updated last year
- ☆60Mar 8, 2022Updated 3 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,792Feb 23, 2026Updated last week
- Run PyTorch in JAX. 🤝☆313Oct 13, 2025Updated 4 months ago
- JAX/Flax implementation of the Hyena Hierarchy☆34Apr 27, 2023Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 4 years ago
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆52Jan 27, 2026Updated last month
- CLU lets you write beautiful training loops in JAX.☆367Updated this week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Updated this week
- ☆273Updated this week
- Exponential families for JAX☆76Feb 20, 2026Updated last week