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 Deep Learning Models in JAX (equinox)☆23Aug 21, 2026Updated last 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
- Minimal Viable Product for an open source, github hosted, python package☆20Oct 22, 2025Updated 10 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆47Jul 10, 2026Updated last month
- 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.
- See and Tell | Assistant to Experience Video Content With Ears☆21Oct 3, 2023Updated 2 years ago
- 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".☆19Oct 27, 2022Updated 3 years ago
- ☆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
- ☆15Sep 24, 2022Updated 3 years ago
- Latest version of IFISS software☆11Dec 27, 2025Updated 8 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
- 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.
- Pytrees + dataclasses ❤️☆76Dec 19, 2025Updated 8 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.☆14Jun 4, 2026Updated 2 months ago
- A GPU-accelerated differentiable fluid simulator written in JAX.☆11Feb 1, 2021Updated 5 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.☆40Jul 30, 2026Updated last month
- 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
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A simple library for scaling up JAX programs☆149Nov 4, 2025Updated 9 months 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
- Transformer creation from scratch using Jax.☆20Dec 8, 2024Updated last year
- singing voice conversion without f0☆22May 10, 2023Updated 3 years ago
- Python-based Blender module for scientific visualization☆12Oct 2, 2018Updated 7 years ago
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Add a tqdm progress bar to your JAX scans and loops.☆130May 22, 2026Updated 3 months ago
- 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
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated last year
- Python port of the Pink Trombone JS code☆18Jan 30, 2022Updated 4 years ago
- Jaxpr Visualisation Tool☆38Dec 22, 2024Updated last year