benmoseley / intro-to-jax-workshopLinks
Introduction to JAX Workshop @ ETH Zurich, 25 June 2024
☆35Updated last month
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 8 months ago
- Example problems in Physics informed neural network in JAX☆80Updated last year
- Efficient Differentiable n-d PDE solvers in JAX.☆39Updated 7 months ago
- ☆32Updated 10 months ago
- ☆17Updated 5 months 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
- Bayesian optimized physics-informed neural network for parameter estimation☆29Updated 6 months ago
- Datasets and code for results presented in the ProbConserv paper☆53Updated 11 months 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
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆31Updated this week
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆31Updated this week
- ☆34Updated last month
- A Python library for training neural ODEs.☆21Updated 3 months ago
- Pseudospectral Kolmogorov Flow Solver☆40Updated last year
- ☆26Updated 10 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆90Updated 2 weeks ago
- Material for workshop and autumn school on scientific machine learning 2023☆20Updated last year
- Two Dimensional Fokker-Planck Solver using Matlab☆19Updated 5 years ago
- ETH Zürich AI in the Sciences and Engineering Master's course 2024☆34Updated 10 months ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆67Updated 4 months ago
- Simulation-Enabled Prediction, Inference, and Analysis: physics-informed statistical learning.☆35Updated last year
- Differentiable interface to FEniCS for JAX☆54Updated 4 years ago
- Learning Green's functions of partial differential equations with deep learning.☆68Updated last year
- Scientific Machine Learning Tutorials☆37Updated 3 years ago
- Bootcamp notebooks☆54Updated last year
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆11Updated 3 weeks ago
- ☆57Updated last year
- Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning☆16Updated last year
- Neural network based solvers for partial differential equations and inverse problems . Implementation of physics-informed neural networks…☆150Updated 4 months ago
- Stochastic Physics-Informed Neural Networks: A Moment-Matching Framework for Learning Hidden Physics within Stochastic Differential Equat…☆14Updated 3 years ago