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)☆25Oct 21, 2026Updated last week
- A PyTorch-based optimizer wrapper for continual learning via selective fine-tuning, guided by the condition number κ (kappa) of model ten…☆13Mar 13, 2026Updated 7 months ago
- JAX models for ML☆12Mar 2, 2026Updated 7 months ago
- ☆36Jul 11, 2017Updated 9 years ago
- Code accompanying the latent-action-priors paper.☆12Mar 5, 2025Updated last year
- 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.
- Bencher makes evaluating black-box problems simple. See https://arxiv.org/abs/2505.21321☆14Oct 1, 2026Updated 3 weeks ago
- Reimplementation using Python of the minimax tilting algorithm by Botev (2016) for simulation and iid sampling of the truncated multivari…☆43Oct 19, 2026Updated last week
- See and Tell | Assistant to Experience Video Content With Ears☆20Oct 3, 2023Updated 3 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated last month
- A general framework for cascade correlation architectures in Python with wrappers to keras, tensorflow and sklearn☆13Jan 7, 2025Updated last year
- This repository contains the code for the paper "Local policy search with Bayesian optimization".☆20Oct 27, 2022Updated 4 years ago
- JAX implementation of RL algorithms and vectorized environments☆50Dec 26, 2023Updated 2 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
- JAX implementations of various deep reinforcement learning algorithms.☆25Feb 2, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- 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
- jax port of snac☆11May 12, 2024Updated 2 years ago
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 7 months ago
- Latest version of IFISS software☆11Dec 27, 2025Updated 10 months ago
- ☆34Oct 5, 2020Updated 6 years ago
- SiTraNo+: An Audio Application For Signal Decompoistion and Time-Scale Modification☆11Dec 12, 2023Updated 2 years ago
- Pytrees + dataclasses ❤️☆76Dec 19, 2025Updated 10 months ago
- A GPU-accelerated differentiable fluid simulator written in JAX.☆11Feb 1, 2021Updated 5 years ago
- Detection of GCIs and prediction of glottal flow from raw speech signals using a fully-convolutional network (FCN)☆13Mar 2, 2020Updated 6 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.
- ☆15Jul 20, 2026Updated 3 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆113Jul 19, 2024Updated 2 years ago
- Simple secrets management for NixOS.☆13Apr 4, 2026Updated 6 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
- An Automized Partial Differential Equation solver based on JAX☆14Jul 9, 2026Updated 3 months ago
- An implementation of PSGD Kron in JAX for distributed training in JAX or Flax☆10Nov 6, 2025Updated 11 months 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
- JAXMg: A multi-GPU linear solver in JAX☆112Oct 22, 2026Updated last week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Single Particle States - Quantum Mechanics in 2D☆11Apr 23, 2021Updated 5 years ago
- Implicit Layer Machine Learning via Deep Equilibrium Networks, O(1) backpropagation with accelerated convergence.☆59Updated this week
- A library of surrogate transport models for tokamak fusion.☆44Oct 19, 2026Updated last week
- ☆10Feb 12, 2019Updated 7 years ago
- Named Tensors for Legible Deep Learning in JAX☆231Sep 1, 2026Updated last month
- Frequent Habits☆29Updated this week
- export JAX to ONNX☆89Updated this week