rezaakb / pinns-jaxLinks
PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.
☆54Updated 10 months ago
Alternatives and similar repositories for pinns-jax
Users that are interested in pinns-jax are comparing it to the libraries listed below
Sorting:
- Example problems in Physics informed neural network in JAX☆80Updated last year
- Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed c…☆114Updated 3 years ago
- This repository contains code, which was used to generate large-scale results in the HINTS paper.☆30Updated 9 months ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆44Updated 10 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆77Updated 2 months ago
- An RL-Gym for Challenge Problems in Data-Driven Modeling and Control of Fluid Dynamics.☆75Updated last week
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆45Updated last year
- Differentiable interface to FEniCS for JAX☆56Updated 4 years ago
- This is the repository for the code used in the ICML23 paper called "Achieving High Accuracy with PINNs via Energy Natural Gradient Desce…☆20Updated 9 months ago
- ☆27Updated last year
- Pseudospectral Kolmogorov Flow Solver☆41Updated last year
- Code for the paper: Solving and Learning Nonlinear PDEs with Gaussian Processes☆39Updated 2 weeks ago
- Spectral Neural Operator☆79Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆45Updated 9 months ago
- Introduction to JAX Workshop @ ETH Zurich, 25 June 2024☆36Updated 3 months ago
- ☆54Updated 2 years ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆70Updated 6 months ago
- In this repository, you will find the different python scripts to train the available models on the AirfRANS dataset proposed at the Neur…☆56Updated 6 months ago
- ☆97Updated 3 years ago
- Example code for paper: Automatic Differentiation to Simultaneously Identify Nonlinear Dynamics and Extract Noise Probability Distributio…☆56Updated 3 years ago
- ☆42Updated 2 years ago
- Applications of PINOs☆129Updated 2 years ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆64Updated 7 months ago
- Transformers for modeling physical systems☆142Updated 2 years ago
- Data-driven reduced order modeling for nonlinear dynamical systems☆84Updated last month
- Update PDEKoopman code to Tensorflow 2☆23Updated 4 years ago
- The unsupervised learning problem trains a diffeomorphic spatio-temporal grid, that registers the output sequence of the PDEs onto a non-…☆19Updated 2 years ago
- Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"☆38Updated 2 years ago
- Methods and code for J. L. Callaham, J. N. Kutz, B. W. Brunton, and S. L. Brunton (2020)☆57Updated 4 years ago
- Links to works on deep learning algorithms for physics problems, TUM-I15 and beyond☆58Updated 4 years ago