A repository for Mixed Precision Training in JAX based on Equinox.
☆16Nov 2, 2025Updated 10 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 5, 2026Updated 2 weeks ago
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆20Updated this week
- Code accompanying the latent-action-priors paper.☆12Mar 5, 2025Updated last year
- ☆36Jul 11, 2017Updated 9 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated last week
- 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.
- 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 3 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
- ☆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 4 years ago
- ☆15Sep 24, 2022Updated 3 years ago
- Slides for Control Theory, Innopolis, 2021☆25Apr 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.
- Latest version of IFISS software☆11Dec 27, 2025Updated 8 months ago
- Training methodologies for autoregressive neural operators/emulators in JAX.☆15Jun 1, 2026Updated 3 months ago
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 6 months ago
- ☆34Oct 5, 2020Updated 5 years ago
- ☆14Jul 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 GPU-accelerated and differentiable library for connectivity modelling at scale.☆14Jun 4, 2026Updated 3 months ago
- 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
- 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.
- 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
- A library of surrogate transport models for tokamak fusion.☆41Sep 9, 2026Updated last week
- Implicit Layer Machine Learning via Deep Equilibrium Networks, O(1) backpropagation with accelerated convergence.☆59Updated this week
- 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 10 months ago
- ☆14Feb 18, 2024Updated 2 years ago
- A Rust binding for Wwise☆18Jan 4, 2023Updated 3 years ago
- 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.
- The Mini Wheelbot is a balancing, reaction wheel unicycle robot designed as a testbed for learning-based control. It is an unstable syste…☆41Aug 26, 2026Updated 3 weeks 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 3 months ago
- a differentiable lensing simulator☆12Jan 18, 2022Updated 4 years ago
- Simulate sygyt style overtone singing (a.k.a. throat singing) by emphasizing MIDI-selected harmonics of an audio signal.☆20Nov 10, 2019Updated 6 years ago
- singing voice conversion without f0☆22May 10, 2023Updated 3 years ago