benmoseley / intro-to-jax-workshopLinks
Introduction to JAX Workshop @ ETH Zurich, 25 June 2024
☆36Updated 3 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:
- Efficient Differentiable n-d PDE solvers in JAX.☆45Updated 8 months ago
- Example problems in Physics informed neural network in JAX☆80Updated last year
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆49Updated 10 months ago
- Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed c…☆114Updated 3 years ago
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆33Updated last week
- ☆17Updated 6 months ago
- Bayesian optimized physics-informed neural network for parameter estimation☆32Updated 7 months ago
- Material for workshop and autumn school on scientific machine learning 2023☆20Updated last year
- Datasets and code for results presented in the ProbConserv paper☆53Updated last year
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆76Updated last month
- JAX-DIPS is a differentiable interfacial PDE solver.☆44Updated 10 months ago
- ☆54Updated 2 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
- Two Dimensional Fokker-Planck Solver using Matlab☆19Updated 5 years ago
- Scientific Machine Learning Tutorials☆38Updated 3 years ago
- Numerical Gaussian Processes for Time-dependent and Non-linear Partial Differential Equations☆69Updated 5 years ago
- ☆35Updated 3 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
- Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop☆20Updated 2 months ago
- Differentiable interface to FEniCS for JAX☆54Updated 4 years ago
- Pseudospectral Kolmogorov Flow Solver☆40Updated last year
- A Python library for training neural ODEs.☆25Updated 5 months ago
- ME 697 - Advanced Scientific Machine Learning☆23Updated 2 months ago
- ☆26Updated last year
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆12Updated 2 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆90Updated last week
- ☆32Updated 11 months ago
- Convolutional Solvers for Partial Differential Equations☆28Updated 5 years ago
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆33Updated this week
- SINDy (Sparse Identification of Nonlinear Dynamics) algorithms☆78Updated 2 years ago