Implementation of Deep Learning Models in JAX (equinox)
☆23Aug 11, 2026Updated last 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:
- Fast, lightweight and parallelised simulation-based inference in JAX.☆23Oct 28, 2025Updated 9 months ago
- A fast, memory-efficient exact MaxSim kernel for late-interaction retrieval and reranking.☆26May 27, 2026Updated 2 months ago
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 7 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆113Jul 19, 2024Updated 2 years ago
- JAX-based linear Einstein-Boltzmann solver for cosmology☆27Nov 13, 2025Updated 9 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- 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 10 months ago
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆34May 12, 2023Updated 3 years ago
- Run PyTorch in JAX. 🤝☆317Oct 13, 2025Updated 10 months ago
- Benchmarking field-level cosmological inference from galaxy surveys.☆16Jul 17, 2025Updated last year
- A Pytorch experimentation with the paper "Tuning computer vision models with task rewards"☆16Jun 28, 2023Updated 3 years ago
- ☆15Apr 14, 2023Updated 3 years ago
- ☆14Aug 8, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 8 months ago
- Euclid Visible Instrument Python package. Includes a simulator and various analysis codes.☆11Mar 25, 2015Updated 11 years ago
- Code for reproducing the paper "Space-Time Continuous PDE Forecasting using Equivariant Neural Fields" (https://arxiv.org/abs/2406.06660)…☆15Nov 18, 2025Updated 9 months ago
- A collection of algorithms and experiment tools for safe sim to real transfer in robotics.☆28May 19, 2026Updated 2 months ago
- ☆30Aug 29, 2024Updated last year
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated 2 years ago
- SO Likelihoods and Theories☆18Aug 4, 2026Updated 2 weeks ago
- Logging library for JAX that is compatible with transformations and primitives such as vmap and scan.☆16Aug 7, 2026Updated last week
- Python module to compute the SED of an AGN following Kubota & Done (2018).☆17Jun 13, 2023Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆51Jun 8, 2026Updated 2 months ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago
- Booster robot descriptions and example motion data☆27Mar 12, 2026Updated 5 months ago
- ☆21Nov 4, 2024Updated last year
- Collecting research materials on neural samplers with diffusion/flow models☆59Jul 4, 2025Updated last year
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆174Nov 11, 2025Updated 9 months ago
- Implementation of GTMP in JAX, RA-L 2025 & ICRA 2026.☆39Apr 28, 2026Updated 3 months ago
- a differentiable lensing simulator☆12Jan 18, 2022Updated 4 years ago
- Implementation of the Kdeep Paper☆22Feb 18, 2021Updated 5 years 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.
- Uncertainty-aware Latent Safety Filters for Avoiding Out-of-Distribution Failures (CoRL 2025)☆31Jun 23, 2026Updated last month
- ☆240Jun 21, 2026Updated last month
- ☆45Aug 26, 2024Updated last year
- Joint trajectory planning for constrained manipulation using the Closed-Chain Affordance framework by Janak Panthi☆14May 23, 2026Updated 2 months ago
- HACMan: Learning Hybrid Actor-Critic Maps for 6D Non-Prehensile Manipulation☆50Oct 14, 2023Updated 2 years ago
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- A collection of robotics problems, for learning and for fun☆16Jul 19, 2026Updated 3 weeks ago