A repository for Mixed Precision Training in JAX based on Equinox.
☆16Nov 2, 2025Updated 9 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 Computer Vision Models in JAX (equinox)☆23Updated this week
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆20Updated this week
- ☆36Jul 11, 2017Updated 9 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆47Jul 10, 2026Updated last month
- This repository contains the code for the paper "Local policy search with Bayesian optimization".☆19Oct 27, 2022Updated 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.
- JAX implementations of various deep reinforcement learning algorithms.☆25Feb 2, 2025Updated last year
- ☆24Jun 9, 2026Updated 2 months 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
- ☆15Sep 24, 2022Updated 3 years ago
- Latest version of IFISS software☆11Dec 27, 2025Updated 7 months ago
- Slides for Control Theory, Innopolis, 2021☆25Apr 8, 2023Updated 3 years ago
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 5 months ago
- ☆34Oct 5, 2020Updated 5 years ago
- 2D Stable Fluids & 3D Stable Fluids using the Fast Fourier Transformation implemented efficiently in Julia.☆16Jan 29, 2022Updated 4 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- 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
- Fast, differentiable, and GPU-accelerated simulation framework for modelling the dynamics of strings, membranes, and plates using modal m…☆14May 13, 2026Updated 2 months ago
- 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
- Implicit Layer Machine Learning via Deep Equilibrium Networks, O(1) backpropagation with accelerated convergence.☆59Jul 29, 2026Updated last week
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated 2 years ago
- A screaming vocal samples dataset.☆13Apr 14, 2023Updated 3 years ago
- A simple library for scaling up JAX programs☆148Nov 4, 2025Updated 9 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.
- ☆14Feb 18, 2024Updated 2 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
- Transformer creation from scratch using Jax.☆20Dec 8, 2024Updated last year
- 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
- 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.
- Python wrapper around the BEZ2018 auditory nerve model☆12Mar 10, 2024Updated 2 years ago
- Python-based Blender module for scientific visualization☆12Oct 2, 2018Updated 7 years ago
- Add a tqdm progress bar to your JAX scans and loops.☆132May 22, 2026Updated 2 months ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- Applet for explaining z transform☆14Apr 26, 2022Updated 4 years ago
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago