benmoseley / intro-to-jax-workshopLinks
Introduction to JAX Workshop @ ETH Zurich, 25 June 2024
☆39Updated 8 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.☆57Updated last year
- Bayesian optimized physics-informed neural network for parameter estimation☆33Updated last year
- Example problems in Physics informed neural network in JAX☆82Updated 2 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated 2 months ago
- Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed c…☆117Updated 3 years ago
- Pseudospectral Kolmogorov Flow Solver☆41Updated 2 years ago
- ☆29Updated last year
- Numerical Gaussian Processes for Time-dependent and Non-linear Partial Differential Equations☆69Updated 5 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…☆26Updated last year
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆94Updated this week
- ☆17Updated last year
- ☆109Updated 4 years ago
- ☆55Updated 2 years ago
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆43Updated last week
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆92Updated last month
- ETH Zürich AI in the Sciences and Engineering Master's course 2024☆42Updated last year
- This codes calculates the dimensionalized POD and uses SINDy from the PySINDy python package to build a data-driven model for it. The cod…☆22Updated 4 years ago
- Codes associated with the manuscript titled "Multi-stage neural networks: Function approximator of machine precision"☆50Updated last year
- A library for dimensionality reduction on spatial-temporal PDE☆70Updated last year
- GCA-ROM is a library which implements graph convolutional autoencoder architecture as a nonlinear model order reduction strategy.☆36Updated last month
- This repository contains code, which was used to generate large-scale results in the HINTS paper.☆35Updated last year
- ☆118Updated 6 years ago
- Example code for paper: Automatic Differentiation to Simultaneously Identify Nonlinear Dynamics and Extract Noise Probability Distributio…☆60Updated 3 years ago
- Neural network based solvers for partial differential equations and inverse problems . Implementation of physics-informed neural networks…☆157Updated 11 months ago
- Practicum on Supervised Learning in Function Spaces☆33Updated 3 years ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆48Updated last year
- Code for "Nonlinear stochastic modeling with Langevin regression" J. L. Callaham, J.-C. Loiseau, G. Rigas, and S. L. Brunton☆26Updated 3 years ago
- ☆54Updated 3 years ago
- Spectral Neural Operator☆78Updated 2 years ago
- The unsupervised learning problem trains a diffeomorphic spatio-temporal grid, that registers the output sequence of the PDEs onto a non-…☆19Updated 3 years ago