AeroJAX: A differentiable, structure-preserving framework for real-time flow simulation, control, and inverse design. Architected for neural operator integration and latent-space acceleration. Built with JAX.
☆319Oct 7, 2026Updated last week
Alternatives and similar repositories for AeroJAX
Users that are interested in AeroJAX are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- GPU-accelerated differentiable sparse linear solver library for JAX☆57Oct 8, 2026Updated last week
- A modular, JAX-based framework for Physics-Informed Neural Networks, designed for scalable PDE-constrained learning with support for doma…☆37Updated this week
- "Neural Emulator Superiority: When Machine Learning for PDEs Surpasses its Training Data" [NeurIPS 2025]☆17May 21, 2026Updated 4 months ago
- Massively-parallelized Navier-Stokes solver for Taylor-Couette flows☆10Jul 6, 2026Updated 3 months ago
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆14May 8, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Training methodologies for autoregressive neural operators/emulators in JAX.☆15Jun 1, 2026Updated 4 months ago
- Efficient GPU-computing simulation platform for differentiable phase field model☆28Apr 14, 2026Updated 6 months ago
- Differentiable Fluid Dynamics Package☆658Aug 28, 2026Updated last month
- P3D☆40Apr 11, 2026Updated 6 months ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆231Jun 1, 2026Updated 4 months ago
- Graphical editor for OpenFOAM blockMeshDict files☆31May 13, 2026Updated 5 months ago
- This repository provides examples of coupling solvers using CoSimIO, including OpenFOAM with Kratos Multiphysics, as well as data exchan…☆50Aug 9, 2024Updated 2 years ago
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆46Nov 21, 2025Updated 10 months ago
- Multi-GPU code for intreface-resolved simulations of multiphase turbulence (triply periodic box and channel). DNS of Navier-Stokes equati…☆33Updated this week
- 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.
- Differentiable Hydrodynamical Simulations☆18Sep 26, 2026Updated 2 weeks ago
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆512Sep 16, 2026Updated 3 weeks ago
- A code for fast, massively-parallel direct numerical simulations (DNS) of canonical flows☆26Oct 1, 2026Updated 2 weeks ago
- A fully-differentiable compressible reacting flow solver based on JAX-AMR☆38Jul 14, 2026Updated 3 months ago
- ☆31Jul 13, 2026Updated 3 months ago
- SParse lINear solvers in JAX☆56Aug 4, 2026Updated 2 months ago
- Pseudo-spectral code for DNS of Homogenous isotropic turbulence. Scalars and particles are also supported.☆12Oct 19, 2023Updated 2 years ago
- ☆15Feb 18, 2024Updated 2 years ago
- Massively-parallelised Navier-Stokes solver with a volume-of-fluid method for bubbly flows☆12Sep 13, 2025Updated last year
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- A simulation node in your JAX workflow.☆90Updated this week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆136May 7, 2026Updated 5 months ago
- JAXMg: A multi-GPU linear solver in JAX☆106Updated this week
- Matrix-free linear algebra in JAX.☆182Updated this week
- A Warp Drive for Neural PDE Solvers☆52Oct 6, 2026Updated last week
- A code for fast, massively-parallel simulations of two-phase flows with heat transfer☆50Aug 31, 2026Updated last month
- FLOW36: A GPU-ready spectral code for DNS of multiphase turbulence based on the phase-field approach.☆40Updated this week
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- Computational Fluid Dynamics in JAX☆965Updated this week
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- A fast, differentiable, JIT-free, debugging-friendly finite element library for PyTorch.☆227Sep 28, 2026Updated 2 weeks ago
- Advanced Finite Difference flow solver with Multiple Resolution and Phase-Field implementations☆26Dec 4, 2025Updated 10 months ago
- differentiable (magneto)hydrodynamics for astrophysics in JAX☆71Oct 7, 2026Updated last week
- ☆10Dec 26, 2025Updated 9 months ago
- Neural Emulator Architectures in JAX.☆28Aug 25, 2026Updated last month
- Video series: building 3D CAD visualization with C++, OpenCascade, Qt, and OpenGL☆18Feb 28, 2026Updated 7 months ago
- Automatic Sparse Differentiation in JAX.☆58Updated this week