rezaakb / pinns-jaxLinks
PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.
☆48Updated 8 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
- Code for the paper: Solving and Learning Nonlinear PDEs with Gaussian Processes☆39Updated 7 months ago
- An RL-Gym for Challenge Problems in Data-Driven Modeling and Control of Fluid Dynamics.☆71Updated this week
- Differentiable interface to FEniCS for JAX☆54Updated 4 years ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆73Updated 2 weeks ago
- Efficient Differentiable n-d PDE solvers in JAX.☆39Updated 7 months ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆44Updated 8 months ago
- Spectral Neural Operator☆77Updated last year
- Introduction to JAX Workshop @ ETH Zurich, 25 June 2024☆35Updated last month
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆67Updated 4 months 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
- This repository contains code, which was used to generate large-scale results in the HINTS paper.☆29Updated 7 months ago
- This is the repository for the code used in the ICML23 paper called "Achieving High Accuracy with PINNs via Energy Natural Gradient Desce…☆19Updated 7 months ago
- ☆17Updated 5 months ago
- ☆97Updated 3 years ago
- Example code for paper: Automatic Differentiation to Simultaneously Identify Nonlinear Dynamics and Extract Noise Probability Distributio…☆55Updated 3 years ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆42Updated last year
- A Python library for training neural ODEs.☆21Updated 3 months ago
- Pseudospectral Kolmogorov Flow Solver☆40Updated last year
- Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"☆38Updated 2 years ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆63Updated 5 months ago
- ☆26Updated 10 months ago
- ☆41Updated 2 years ago
- Methods and code for J. L. Callaham, J. N. Kutz, B. W. Brunton, and S. L. Brunton (2020)☆56Updated 4 years ago
- Transformers for modeling physical systems☆140Updated last year
- ☆29Updated 2 years ago
- Material for workshop and autumn school on scientific machine learning 2023☆20Updated last year
- Update PDEKoopman code to Tensorflow 2☆23Updated 4 years ago
- Data-driven reduced order modeling for nonlinear dynamical systems☆30Updated 2 weeks ago
- Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed c…☆115Updated 3 years ago