Ceyron / pdequinoxView external linksLinks
Neural Emulator Architectures in JAX.
☆23Nov 5, 2024Updated last year
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.☆53Updated this week
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆22Oct 7, 2023Updated 2 years ago
- A minimal JAX library for connectivity modelling at scale☆11Jan 27, 2026Updated 2 weeks ago
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆29May 30, 2025Updated 8 months ago
- Simple differentiable approximate ocean models built with JAX.☆16Oct 13, 2023Updated 2 years ago
- ☆14Feb 18, 2024Updated last year
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated 7 months ago
- Special functions in Jax☆16Mar 1, 2025Updated 11 months ago
- 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 6 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Updated this week
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- TorchFSM: Fourier Spectral Method with PyTorch☆53Jan 30, 2026Updated 2 weeks 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
- A simple JAX solver for the 2D Euler equations for benchmarking☆19Nov 3, 2025Updated 3 months ago
- [ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch☆72Nov 21, 2025Updated 2 months ago
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆32Apr 19, 2025Updated 9 months ago
- Flood Mapping Intercomparison☆15Nov 5, 2025Updated 3 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆541Updated this week
- Jaxpr Visualisation Tool☆35Dec 22, 2024Updated last year
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆39Nov 21, 2025Updated 2 months ago
- Astrostatistics and Machine Learning class for the MSc degree in Astrophysics at the University of Milan-Bicocca (Italy)☆15Aug 30, 2025Updated 5 months ago
- Introduction to JAX Workshop @ ETH Zurich, 25 June 2024☆39Apr 16, 2025Updated 9 months ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last week
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆68Jan 1, 2025Updated last year
- Spectral Neural Operator☆79Dec 20, 2023Updated 2 years ago
- Intuitive symbolic interface for constrained optimization problems. Write natural Python, get automatic gradients and solvers.☆24Updated this week
- Codes for Numerical Analysis course☆39Apr 22, 2025Updated 9 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 7, 2026Updated last week
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆53Jul 15, 2025Updated 6 months ago
- MIT PhD Thesis template☆11May 22, 2025Updated 8 months ago
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 2 years ago
- Latest version of IFISS software☆11Dec 27, 2025Updated last month