A repository for Mixed Precision Training in JAX based on Equinox.
☆16Nov 2, 2025Updated 11 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:
- Implementation of Deep Learning Models in JAX (equinox)☆25Sep 24, 2026Updated 2 weeks ago
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆21Updated this week
- Code accompanying the latent-action-priors paper.☆12Mar 5, 2025Updated last year
- ☆36Jul 11, 2017Updated 9 years ago
- Minimal Viable Product for an open source, github hosted, python package☆20Oct 22, 2025Updated 11 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated 3 weeks ago
- See and Tell | Assistant to Experience Video Content With Ears☆20Oct 3, 2023Updated 3 years ago
- A general framework for cascade correlation architectures in Python with wrappers to keras, tensorflow and sklearn☆13Jan 7, 2025Updated last year
- JAX implementation of RL algorithms and vectorized environments☆50Dec 26, 2023Updated 2 years ago
- JAX implementations of various deep reinforcement learning algorithms.☆25Feb 2, 2025Updated last year
- ☆24Jun 9, 2026Updated 4 months 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
- 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 4 years ago
- ☆15Sep 24, 2022Updated 4 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.
- Latest version of IFISS software☆11Dec 27, 2025Updated 9 months ago
- Training methodologies for autoregressive neural operators/emulators in JAX.☆15Jun 1, 2026Updated 4 months ago
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 7 months ago
- ☆34Oct 5, 2020Updated 6 years ago
- ☆15Jul 20, 2026Updated 2 months ago
- Pytrees + dataclasses ❤️☆76Dec 19, 2025Updated 9 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☆11Jun 6, 2022Updated 4 years ago
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆15Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- A GPU-accelerated differentiable fluid simulator written in JAX.☆11Feb 1, 2021Updated 5 years ago
- 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
- BAD-VAE: A VAE framework for unsupervised disentanglement of sequential data☆12May 25, 2022Updated 4 years ago
- Website for the book "The Elements of Differentiable Programming".☆15Jul 2, 2025Updated last year
- 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☆149Nov 4, 2025Updated 11 months ago
- ☆15Feb 18, 2024Updated 2 years ago
- A Rust binding for Wwise☆18Jan 4, 2023Updated 3 years ago
- The Mini Wheelbot is a balancing, reaction wheel unicycle robot designed as a testbed for learning-based control. It is an unstable syste…☆43Sep 30, 2026Updated last week
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- 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
- A framework for overviewing the performance of F0 estimators☆19Sep 10, 2016Updated 10 years ago
- a differentiable lensing simulator☆12Jan 18, 2022Updated 4 years ago
- Transformer creation from scratch using Jax.☆20Dec 8, 2024Updated last year
- This is the code repository for the DuSt-MPC paper, published at Robotics: Science and Systems (RSS) 2021.☆31Jul 16, 2021Updated 5 years ago
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 3 years ago