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.
☆306Aug 13, 2026Updated last month
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☆53Aug 30, 2026Updated 3 weeks ago
- A modular, JAX-based framework for Physics-Informed Neural Networks, designed for scalable PDE-constrained learning with support for doma…☆36Sep 16, 2026Updated last 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 2 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.
- Efficient GPU-computing simulation platform for differentiable phase field model☆28Apr 14, 2026Updated 5 months ago
- Differentiable Fluid Dynamics Package☆645Aug 28, 2026Updated 3 weeks ago
- P3D☆40Apr 11, 2026Updated 5 months ago
- computational physics (Chungbuk National University, Korea)☆10May 26, 2022Updated 4 years ago
- Graphical editor for OpenFOAM blockMeshDict files☆31May 13, 2026Updated 4 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…☆33Jun 25, 2026Updated 3 months ago
- Differentiable Hydrodynamical Simulations☆17Jul 17, 2026Updated 2 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆508Sep 16, 2026Updated last week
- A code for fast, massively-parallel direct numerical simulations (DNS) of canonical flows☆25Updated this week
- A fully-differentiable compressible reacting flow solver based on JAX-AMR☆38Jul 14, 2026Updated 2 months ago
- ☆29Jul 13, 2026Updated 2 months ago
- SParse lINear solvers in JAX☆54Aug 4, 2026Updated last month
- Pseudo-spectral code for DNS of Homogenous isotropic turbulence. Scalars and particles are also supported.☆12Oct 19, 2023Updated 2 years ago
- ☆14Feb 18, 2024Updated 2 years ago
- Massively-parallelised Navier-Stokes solver with a volume-of-fluid method for bubbly flows☆11Sep 13, 2025Updated last year
- A simulation node in your JAX workflow.☆89Aug 3, 2026Updated last month
- 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.
- JAXMg: A multi-GPU linear solver in JAX☆71Updated this week
- ☆61Apr 20, 2026Updated 5 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆135May 7, 2026Updated 4 months ago
- Matrix-free linear algebra in JAX.☆180Sep 2, 2026Updated 3 weeks ago
- A Warp Drive for Neural PDE Solvers☆50Jun 3, 2026Updated 3 months ago
- A code for fast, massively-parallel simulations of two-phase flows with heat transfer☆50Aug 31, 2026Updated 3 weeks ago
- FLOW36: A GPU-ready spectral code for DNS of multiphase turbulence based on the phase-field approach.☆40Feb 11, 2026Updated 7 months ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- Computational Fluid Dynamics in JAX☆964Updated 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.☆224Sep 17, 2026Updated last week
- Advanced Finite Difference flow solver with Multiple Resolution and Phase-Field implementations☆26Dec 4, 2025Updated 9 months ago
- differentiable (magneto)hydrodynamics for astrophysics in JAX☆71Sep 2, 2026Updated 3 weeks ago
- ☆10Dec 26, 2025Updated 8 months ago
- ☆23Updated this week
- a Thermal Lattice Boltzmann code for concentrated emulsions with FINite-size Droplets☆12Feb 23, 2022Updated 4 years ago
- Neural Emulator Architectures in JAX.☆27Aug 25, 2026Updated last month