Physics-informed neural networks in JAX — multi-GPU training, advanced PINN algorithms, and 16 benchmark examples
☆35Aug 12, 2026Updated 2 months ago
Alternatives and similar repositories for jaxpi2
Users that are interested in jaxpi2 are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- This is the official implementation of "Deep Fuzzy Physics-Informed Neural Networks for Forward and Inverse PDE Problems" (Neural Network…☆30Oct 14, 2025Updated last year
- ☆46Mar 4, 2025Updated last year
- Official Implementation of "AIVT: Inference of turbulent thermal convection from measured 3D velocity data by physics-informed Kolmogorov…☆16Sep 19, 2026Updated last month
- Scale-PINN: Learning Efficient Physics-Informed Neural Networks Through Sequential Correction☆31Feb 27, 2026Updated 7 months ago
- This is the official implementation of "CoPINN: Cognitive Physics-informed Neural Network" (ICML 2025, Spotlight)☆27Apr 12, 2026Updated 6 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Replication with PyTorch of ''Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involv…☆41Oct 1, 2026Updated 3 weeks ago
- Code for "Dual Cone Gradient Descent for Physics-Informed Neural Networks☆14Dec 6, 2025Updated 10 months ago
- Enhancing the convergence speed by 2x and improving the training success of Physics-Informed Neural Networks (PINNs).☆14Oct 14, 2024Updated 2 years ago
- PF-PINNs: physics-informed neural networks framework for solving coupled Allen-Cahn and Cahn-Hilliard phase field equations☆38Feb 18, 2025Updated last year
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆56Updated this week
- ☆76Oct 5, 2026Updated 2 weeks ago
- Neural topology optimization via active learning for efficient channel design in turbulent mass transfer☆17Oct 9, 2025Updated last year
- A modular, JAX-based framework for Physics-Informed Neural Networks, designed for scalable PDE-constrained learning with support for doma…☆37Oct 10, 2026Updated 2 weeks ago
- Code accompanying "Inverse-Dirichlet Weighting Enables Reliable Training of Physics Informed Neural Networks", Maddu et al., 2021☆14Nov 3, 2021Updated 4 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A minimal AI agent built from scratch — no agent framework, just Python, the OpenAI SDK, and a `while` loop. Accompanies the Medium artic…☆37Aug 31, 2026Updated last month
- ☆15Mar 13, 2026Updated 7 months ago
- Staggered HARd-constrained Physics-Informed Neural Networks for Phase-field modelling of corrosion☆18Feb 19, 2025Updated last year
- [NeurIPS 2023 Spotlight] Separable Physics-Informed Neural Networks☆95Oct 25, 2023Updated 3 years ago
- ☆17Apr 18, 2024Updated 2 years ago
- ☆15Dec 13, 2022Updated 3 years ago
- XPINN code written in TensorFlow 2☆28Feb 1, 2023Updated 3 years ago
- This repository contains the code for coupling FEniCS with Physics-Informed DeepONet☆23Mar 28, 2026Updated 6 months ago
- The lid-driven cavity is a popular problem within the field of computational fluid dynamics (CFD) for validating computational methods. I…☆16Dec 14, 2021Updated 4 years 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.
- AI-CFD hybrid solver.☆20Apr 21, 2026Updated 6 months ago
- Differentiable numerical solvers and scientific machine learning on one JAX substrate; you write the math, it stays differentiable end to…☆32Updated this week
- Physics Informed Learning System. Official code for "Preconditioned Physics-Informed Neural Operator Training."☆17Sep 30, 2026Updated 3 weeks ago
- Speeding up and reducing memory usage for scientific machine learning via mixed precision☆12Dec 27, 2024Updated last year
- Physics Informed Neural Network (PINN) for the 2D Navier-Stokes equation☆44Jun 24, 2022Updated 4 years ago
- ☆55Aug 19, 2025Updated last year
- ☆15Jun 30, 2025Updated last year
- Neural force field learning toolkit☆14Jul 17, 2026Updated 3 months ago
- PINNs-MPF is a comprehensive framework designed for simulating interface dynamics using Physics-Informed Neural Networks (PINNs). Leverag…☆24Aug 30, 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.
- Folax (Finite Operator Learning with JAX) is a framework for solving and optimizing PDEs by integrating machine learning with numerical m…☆29Updated this week
- SK-PINN: Accelerated physics-informed deep learning by smoothing kernel gradients☆24Apr 7, 2025Updated last year
- ☆19Nov 24, 2025Updated 11 months ago
- PINN paper that will be submitted to Journal of Computational Science☆13Sep 11, 2026Updated last month
- Differentiable N-body code (JAX)☆11May 8, 2025Updated last year
- Some useful tools for mesh processing☆12Sep 9, 2025Updated last year
- a machine learning tool to detect and classify bee buzzes in audio☆17Updated this week