benmoseley / intro-to-jax-workshop
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:
- Efficient Differentiable n-d PDE solvers in JAX.☆37Updated 6 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆70Updated 6 months ago
- A library for dimensionality reduction on spatial-temporal PDE☆66Updated last year
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆48Updated 8 months ago
- ☆32Updated 9 months ago
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs with JAX 📓 Check out our various notebooks to get started ⚠️ Mirror repos…☆29Updated last week
- ☆26Updated 10 months ago
- Material for workshop and autumn school on scientific machine learning 2023☆20Updated last year
- ☆17Updated 4 months ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆141Updated last month
- Example problems in Physics informed neural network in JAX☆80Updated last year
- Spectral Neural Operator☆78Updated last year
- Pseudospectral Kolmogorov Flow Solver☆39Updated last year
- Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning☆17Updated last year
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆87Updated last week
- A Python library for training neural ODEs.☆21Updated 3 months ago
- Two Dimensional Fokker-Planck Solver using Matlab☆19Updated 5 years ago
- Practicum on Supervised Learning in Function Spaces☆33Updated 3 years ago
- Bayesian optimized physics-informed neural network for parameter estimation☆28Updated 5 months ago
- Differentiable interface to FEniCS for JAX☆53Updated 3 years 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
- GCA-ROM is a library which implements graph convolutional autoencoder architecture as a nonlinear model order reduction strategy.☆36Updated last month
- Simulation-Enabled Prediction, Inference, and Analysis: physics-informed statistical learning.☆35Updated last year
- JAX-DIPS is a differentiable interfacial PDE solver.☆41Updated 8 months ago
- ☆93Updated 3 years ago
- [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.☆41Updated last year
- Code for the paper "Generative AI for fast and accurate statistical computation of fluids"☆28Updated 2 months ago
- An interpretable data-driven framework for building generative reduced order models with embedded uncertainty quantification☆32Updated 2 months ago
- This repository contains code, which was used to generate large-scale results in the HINTS paper.☆28Updated 6 months ago
- Kolmogorov-Arnold Networks built on JAX☆77Updated last month