benmoseley / intro-to-jax-workshopLinks
Introduction to JAX Workshop @ ETH Zurich, 25 June 2024
☆36Updated 4 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:
- Example problems in Physics informed neural network in JAX☆80Updated 2 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆47Updated 9 months ago
- Bayesian optimized physics-informed neural network for parameter estimation☆32Updated 9 months ago
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆54Updated 11 months ago
- ☆27Updated last year
- Neural network based solvers for partial differential equations and inverse problems . Implementation of physics-informed neural networks…☆153Updated 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…☆21Updated 10 months ago
- Pseudospectral Kolmogorov Flow Solver☆40Updated last year
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆12Updated 3 months ago
- Generalized sparse regression for continuous and discrete data☆11Updated 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
- ETH Zürich AI in the Sciences and Engineering Master's course 2024☆39Updated last year
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆151Updated 4 months ago
- Codes associated with the manuscript titled "Multi-stage neural networks: Function approximator of machine precision"☆46Updated last year
- ☆17Updated 8 months ago
- ☆99Updated 3 years ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆80Updated 3 months ago
- GCA-ROM is a library which implements graph convolutional autoencoder architecture as a nonlinear model order reduction strategy.☆36Updated 2 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆90Updated 2 weeks ago
- Bootcamp notebooks☆58Updated 3 weeks 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
- Applications of PINOs☆132Updated 2 years ago
- Gaussian process-based interpretable latent space dynamics identification through deep autoencoder☆34Updated last week
- Practicum on Supervised Learning in Function Spaces☆33Updated 3 years ago
- A library for dimensionality reduction on spatial-temporal PDE☆66Updated last year
- Datasets and code for results presented in the ProbConserv paper☆53Updated last year
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆35Updated 3 weeks ago
- Spectral Neural Operator☆78Updated last year
- ☆54Updated 2 years ago
- Using NVIDIA modulus for airfoil optimizations at different angles.☆23Updated 2 years ago