benmoseley / intro-to-jax-workshop
Introduction to JAX Workshop @ ETH Zurich, 25 June 2024
☆25Updated 7 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
- ☆22Updated 7 months ago
- XPINN code written in TensorFlow 2☆27Updated 2 years ago
- ☆52Updated 2 years ago
- Physics-informed neural networks☆14Updated 4 years ago
- This repository contains code, which was used to generate large-scale results in the HINTS paper.☆25Updated 3 months ago
- Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning☆17Updated last year
- ☆88Updated 3 years ago
- ETH Zürich AI in the Sciences and Engineering Master's course 2024☆29Updated 6 months ago
- A library for dimensionality reduction on spatial-temporal PDE☆63Updated 10 months ago
- Multifidelity DeepONet☆27Updated last year
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆59Updated 3 months ago
- Discontinuity Computing Using Physics-Informed Neural Network☆22Updated 9 months ago
- ☆49Updated last year
- Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed c…☆114Updated 2 years ago
- A python script to solve the Cahn-Hilliard equation using an implicit pseudospectral method☆37Updated 6 months ago
- A Python library for training neural ODEs.☆20Updated this week
- ☆37Updated 2 years ago
- A sequential DeepONet model implementation that uses a recurrent neural network (GRU and LSTM) in the branch and a feed-forward neural ne…☆12Updated 9 months ago
- Python codes for Locally Adaptive Activation Function (LAAF) used in deep neural networks. Please cite this work as "A D Jagtap, K Kawa…☆38Updated 2 years ago
- ☆32Updated 6 months ago
- DeepGreen network written in Tensorflow 2☆30Updated 3 years ago
- GCA-ROM is a library which implements graph convolutional autoencoder architecture as a nonlinear model order reduction strategy.☆31Updated 7 months ago
- Convolutional Solvers for Partial Differential Equations☆28Updated 4 years ago
- Fourier Neural Operators to solve for Allen Cahn PDE equations☆17Updated 3 years ago
- Pseudospectral Kolmogorov Flow Solver☆37Updated last year
- Example problems in Physics informed neural network in JAX☆78Updated last year
- Two Dimensional Fokker-Planck Solver using Matlab☆19Updated 5 years ago
- ☆11Updated 2 years ago
- ☆10Updated last year
- ☆12Updated last year