A repository for Mixed Precision Training in JAX based on Equinox.
☆16Nov 2, 2025Updated 8 months ago
Alternatives and similar repositories for mixed_precision_for_JAX
Users that are interested in mixed_precision_for_JAX are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Code accompanying the latent-action-priors paper.☆12Mar 5, 2025Updated last year
- ☆36Jul 11, 2017Updated 9 years ago
- Reimplementation using Python of the minimax tilting algorithm by Botev (2016) for simulation and iid sampling of the truncated multivari…☆41Feb 2, 2023Updated 3 years ago
- Minimal Viable Product for an open source, github hosted, python package☆20Oct 22, 2025Updated 8 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆47Jul 10, 2026Updated last week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- See and Tell | Assistant to Experience Video Content With Ears☆21Oct 3, 2023Updated 2 years ago
- JAX implementation of RL algorithms and vectorized environments☆50Dec 26, 2023Updated 2 years ago
- This repository contains the code for the paper "Local policy search with Bayesian optimization".☆19Oct 27, 2022Updated 3 years ago
- JAX implementations of various deep reinforcement learning algorithms.☆25Feb 2, 2025Updated last year
- ☆24Jun 9, 2026Updated last month
- 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
- ☆17Apr 9, 2023Updated 3 years ago
- This repository is created to share my list of sources that I usually find through the internet (Twitter, Google, Reddit, friends' recomm…☆21Sep 10, 2022Updated 3 years ago
- Latest version of IFISS software☆11Dec 27, 2025Updated 6 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Training methodologies for autoregressive neural operators/emulators in JAX.☆13Jun 1, 2026Updated last month
- ☆34Oct 5, 2020Updated 5 years ago
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 4 months ago
- ☆13Aug 2, 2025Updated 11 months ago
- Pytrees + dataclasses ❤️☆76Dec 19, 2025Updated 7 months ago
- 2D Stable Fluids & 3D Stable Fluids using the Fast Fourier Transformation implemented efficiently in Julia.☆16Jan 29, 2022Updated 4 years ago
- A simple pseudo-spectral solver for the Direct Numerical Simulation (DNS) of the 3D Taylor-Green Vortex in the Julia programming language☆10Jun 6, 2022Updated 4 years ago
- A GPU-accelerated and differentiable library for landscape genetics and ecological connectivity modelling.☆13Jun 4, 2026Updated last month
- A variational auto-encoder (VAE) framework with a new type of prior "Variational Mixture of Posteriors" prior, or VampPrior for short.☆10Apr 7, 2021Updated 5 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- BAD-VAE: A VAE framework for unsupervised disentanglement of sequential data☆12May 25, 2022Updated 4 years ago
- A library of surrogate transport models for tokamak fusion.☆38Jul 10, 2026Updated last week
- Implicit Layer Machine Learning via Deep Equilibrium Networks, O(1) backpropagation with accelerated convergence.☆59Jul 14, 2026Updated last week
- Scripts, Files, and Resources for Constructing a Large-scale Dataset of Blackbox Effects for Timbre Transfer☆16Feb 4, 2023Updated 3 years ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated 2 years ago
- A simple library for scaling up JAX programs☆148Nov 4, 2025Updated 8 months ago
- A screaming vocal samples dataset.☆13Apr 14, 2023Updated 3 years ago
- ☆14Feb 18, 2024Updated 2 years ago
- A Rust binding for Wwise☆17Jan 4, 2023Updated 3 years 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.
- export JAX to ONNX☆88Jun 25, 2026Updated 3 weeks ago
- Graphical Processing Unit algorithms for scientific computing☆15Jan 21, 2009Updated 17 years ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- "Neural Emulator Superiority: When Machine Learning for PDEs Surpasses its Training Data" [NeurIPS 2025]☆17May 21, 2026Updated 2 months ago
- A framework for overviewing the performance of F0 estimators☆19Sep 10, 2016Updated 9 years ago
- a differentiable lensing simulator☆12Jan 18, 2022Updated 4 years ago