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.
☆285Aug 13, 2026Updated 3 weeks ago
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☆50Updated this week
- A modular, JAX-based framework for Physics-Informed Neural Networks, designed for scalable PDE-constrained learning with support for doma…☆34Updated this week
- "Neural Emulator Superiority: When Machine Learning for PDEs Surpasses its Training Data" [NeurIPS 2025]☆17May 21, 2026Updated 3 months ago
- Massively-parallelized Navier-Stokes solver for Taylor-Couette flows☆10Jul 6, 2026Updated last month
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆14May 8, 2025Updated last year
- 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.
- Training methodologies for autoregressive neural operators/emulators in JAX.☆15Jun 1, 2026Updated 3 months ago
- Efficient GPU-computing simulation platform for differentiable phase field model☆28Apr 14, 2026Updated 4 months ago
- P3D☆39Apr 11, 2026Updated 4 months ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆231Jun 1, 2026Updated 3 months ago
- computational physics (Chungbuk National University, Korea)☆10May 26, 2022Updated 4 years 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
- Differentiable Fluid Dynamics Package☆636Aug 28, 2026Updated last week
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆45Nov 21, 2025Updated 9 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 2 months 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.
- Differentiable Hydrodynamical Simulations☆17Jul 17, 2026Updated last month
- A fully-differentiable compressible reacting flow solver based on JAX-AMR☆38Jul 14, 2026Updated last month
- ☆29Jul 13, 2026Updated last month
- SParse lINear solvers in JAX☆53Aug 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 11 months ago
- A simulation node in your JAX workflow.☆89Aug 3, 2026Updated last month
- JAXMg: A multi-GPU linear solver in JAX☆65Updated this week
- 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.
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆503May 29, 2026Updated 3 months ago
- ☆59Apr 20, 2026Updated 4 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆135May 7, 2026Updated 3 months ago
- Matrix-free linear algebra in JAX.☆180Updated this week
- A Warp Drive for Neural PDE Solvers☆48Jun 3, 2026Updated 3 months ago
- A code for fast, massively-parallel simulations of two-phase flows with heat transfer☆49Updated this week
- FLOW36: A GPU-ready spectral code for DNS of multiphase turbulence based on the phase-field approach.☆40Feb 11, 2026Updated 6 months ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- Computational Fluid Dynamics in JAX☆962Jul 8, 2026Updated last month
- 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.☆222Aug 4, 2026Updated last month
- 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☆71Updated this week
- ☆10Dec 26, 2025Updated 8 months ago
- ☆23Jun 12, 2026Updated 2 months ago
- 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 week