Implementation of Deep Learning Models in JAX (equinox)
☆25Sep 24, 2026Updated this week
Alternatives and similar repositories for Equimo
Users that are interested in Equimo are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A repository for Mixed Precision Training in JAX based on Equinox.☆16Nov 2, 2025Updated 10 months ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆23Oct 28, 2025Updated 10 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated 2 weeks ago
- A fast, memory-efficient exact MaxSim kernel for late-interaction retrieval and reranking.☆28May 27, 2026Updated 4 months ago
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 8 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A Python package of computer vision models for the Equinox ecosystem.☆113Jul 19, 2024Updated 2 years ago
- Jax implementation of a flexible audio loudness meter in Python with implementation of ITU-R BS.1770-4 loudness algorithm☆13Jan 29, 2025Updated last year
- JAX-based linear Einstein-Boltzmann solver for cosmology☆27Nov 13, 2025Updated 10 months ago
- ☆27Mar 5, 2025Updated last year
- Python API for the EAGLE cosmological simulation database.☆12Mar 15, 2023Updated 3 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated last year
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆34May 12, 2023Updated 3 years ago
- ☆19Apr 22, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆15Sep 24, 2022Updated 4 years ago
- Run PyTorch in JAX. 🤝☆318Oct 13, 2025Updated 11 months ago
- Benchmarking field-level cosmological inference from galaxy surveys.☆16Jul 17, 2025Updated last year
- ☆15Apr 14, 2023Updated 3 years ago
- ☆14Aug 8, 2023Updated 3 years ago
- Euclid Visible Instrument Python package. Includes a simulator and various analysis codes.☆11Mar 25, 2015Updated 11 years ago
- ☆25Jan 22, 2025Updated last year
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- Code for reproducing the paper "Space-Time Continuous PDE Forecasting using Equivariant Neural Fields" (https://arxiv.org/abs/2406.06660)…☆16Nov 18, 2025Updated 10 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Distributed algorithms in Lean 4☆18Jul 28, 2026Updated last month
- Fast, differentiable, and GPU-accelerated simulation framework for modelling the dynamics of strings, membranes, and plates using modal m…☆14May 13, 2026Updated 4 months ago
- A collection of algorithms and experiment tools for safe sim to real transfer in robotics.☆31May 19, 2026Updated 4 months ago
- Extra modules that nobody needs.☆23Jun 14, 2026Updated 3 months ago
- ☆30Aug 29, 2024Updated 2 years ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated 2 years ago
- SO Likelihoods and Theories☆18Updated this week
- Repo for Bring Your Own Vision-Language-Action (VLA) model, arxiv 2024☆39Jan 22, 2025Updated last year
- A unifying framework for linear temporal logics in Lean☆19Nov 7, 2025Updated 10 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Logging library for JAX that is compatible with transformations and primitives such as vmap and scan.☆16Sep 1, 2026Updated 3 weeks ago
- Python module to compute the SED of an AGN following Kubota & Done (2018).☆17Jun 13, 2023Updated 3 years ago
- A screaming vocal samples dataset.☆13Apr 14, 2023Updated 3 years ago
- Nixpkgs CUDA infra [maintainer=@GaetanLepage, @SomeoneSerge]☆21Updated this week
- Graphical Processing Unit algorithms for scientific computing☆15Jan 21, 2009Updated 17 years ago
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago