benmoseley / intro-to-jax-workshopLinks
Introduction to JAX Workshop @ ETH Zurich, 25 June 2024
☆36Updated 2 months ago
Alternatives and similar repositories for intro-to-jax-workshop
Users that are interested in intro-to-jax-workshop are comparing it to the libraries listed below
Sorting:
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆48Updated 9 months ago
- 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…☆115Updated 3 years ago
- ☆17Updated 6 months ago
- ETH Zürich AI in the Sciences and Engineering Master's course 2024☆38Updated 11 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆40Updated 7 months ago
- Symbolic Identification of Non-linear Dynamics. The method generalizes the SINDy algorithm by combining sparse and genetic-programming-ba…☆76Updated 2 years ago
- Bayesian optimized physics-informed neural network for parameter estimation☆30Updated 7 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆75Updated last month
- ☆30Updated last year
- ☆32Updated 11 months ago
- Example code for paper: Automatic Differentiation to Simultaneously Identify Nonlinear Dynamics and Extract Noise Probability Distributio…☆56Updated 3 years ago
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆31Updated this week
- This is the repository for the code used in the ICML23 paper called "Achieving High Accuracy with PINNs via Energy Natural Gradient Desce…☆20Updated 8 months ago
- Neural network based solvers for partial differential equations and inverse problems . Implementation of physics-informed neural networks…☆151Updated 5 months ago
- Kolmogorov-Arnold Networks built on JAX☆82Updated 2 months ago
- TorchFSM: Fourier Spectral Method with PyTorch☆44Updated 2 weeks 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
- ☆35Updated 2 months ago
- ME 697 - Advanced Scientific Machine Learning☆22Updated last month
- Numerical Gaussian Processes for Time-dependent and Non-linear Partial Differential Equations☆69Updated 5 years ago
- This codes calculates the dimensionalized POD and uses SINDy from the PySINDy python package to build a data-driven model for it. The cod…☆20Updated 4 years ago
- An RL-Gym for Challenge Problems in Data-Driven Modeling and Control of Fluid Dynamics.☆75Updated this week
- Material for workshop and autumn school on scientific machine learning 2023☆20Updated last year
- ☆97Updated 3 years ago
- Differentiable interface to FEniCS for JAX☆54Updated 4 years ago
- Practicum on Supervised Learning in Function Spaces☆33Updated 3 years ago
- Bootcamp notebooks☆55Updated last year
- Physics-informed neural networks☆15Updated 4 years ago
- Learning Green's functions of partial differential equations with deep learning.☆69Updated last year