Neural Emulator Architectures in JAX.
☆23Feb 20, 2026Updated 2 weeks ago
Alternatives and similar repositories for pdequinox
Users that are interested in pdequinox are comparing it to the libraries listed below
Sorting:
- Efficient Differentiable n-d PDE Solvers in JAX.☆155Updated this week
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆99Nov 11, 2025Updated 3 months ago
- A minimal JAX library for connectivity modelling at scale☆11Jan 27, 2026Updated last month
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆29May 30, 2025Updated 9 months ago
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated 8 months ago
- Special functions in Jax☆17Mar 1, 2025Updated last year
- Transformer creation from scratch using Jax.☆19Dec 8, 2024Updated last year
- Chrome browser extension that adds customizable keyboard shortcuts to the Overleaf online latex editor.☆13Jun 22, 2024Updated last year
- Pseudospectral Kolmogorov Flow Solver☆42Oct 24, 2023Updated 2 years ago
- Finite difference tools in JAX☆23Feb 4, 2024Updated 2 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19May 25, 2024Updated last year
- Gaussian Mixture Models in Jax☆24Jul 14, 2025Updated 7 months ago
- ☆36Jul 11, 2017Updated 8 years ago
- Connect to Slurm Compute Nodes Directly from VSCode. Do your debugging and development easily.☆28Oct 8, 2024Updated last year
- GCM Physics written in JAX☆85Updated this week
- a real-time N-body simulation with the Barnes-Hut algorithm and CUDA☆30Jan 1, 2025Updated last year
- A simple JAX solver for the 2D Euler equations for benchmarking☆19Nov 3, 2025Updated 4 months ago
- NeuralGCM's differentiable dycore☆317Feb 27, 2026Updated last week
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆34Apr 19, 2025Updated 10 months ago
- [ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch☆76Nov 21, 2025Updated 3 months ago
- Jaxpr Visualisation Tool☆36Dec 22, 2024Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆546Feb 16, 2026Updated 2 weeks ago
- Introduction to JAX Workshop @ ETH Zurich, 25 June 2024☆42Apr 16, 2025Updated 10 months ago
- Astrostatistics and Machine Learning class for the MSc degree in Astrophysics at the University of Milan-Bicocca (Italy)☆15Aug 30, 2025Updated 6 months ago
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆41Nov 21, 2025Updated 3 months ago
- Flood Mapping Intercomparison☆16Nov 5, 2025Updated 4 months ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆95Feb 27, 2026Updated last week
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Feb 10, 2026Updated 3 weeks ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆70Jan 1, 2025Updated last year
- Matrix-free linear algebra in JAX.☆160Dec 2, 2025Updated 3 months ago
- Spectral Neural Operator☆79Dec 20, 2023Updated 2 years ago
- A differentiable finite element analysis solver for structural optimization based on JAX☆40Dec 22, 2025Updated 2 months ago
- Intuitive symbolic interface for constrained optimization problems. Write natural Python, get automatic gradients and solvers.☆24Feb 17, 2026Updated 2 weeks ago
- Codes for Numerical Analysis course☆39Apr 22, 2025Updated 10 months ago
- A hands-on tutorial on supervised learning with Gaussian processes☆37Nov 22, 2019Updated 6 years ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Feb 27, 2026Updated last week