Neural Emulator Architectures in JAX.
☆28Aug 25, 2026Updated last month
Alternatives and similar repositories for pdequinox
Users that are interested in pdequinox are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Training methodologies for autoregressive neural operators/emulators in JAX.☆15Jun 1, 2026Updated 4 months ago
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆46Nov 21, 2025Updated 10 months ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆231Jun 1, 2026Updated 4 months ago
- Official implementation of "Guiding diffusion models to reconstruct flow fields from sparse data"☆21Jan 20, 2026Updated 8 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆109Jun 1, 2026Updated 4 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆36May 30, 2025Updated last year
- P3D☆40Apr 11, 2026Updated 6 months ago
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 3 years ago
- "Neural Emulator Superiority: When Machine Learning for PDEs Surpasses its Training Data" [NeurIPS 2025]☆17May 21, 2026Updated 4 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆136May 7, 2026Updated 5 months ago
- Coordinates in JAX☆38Updated this week
- Differentiable Hydrodynamical Simulations☆18Sep 26, 2026Updated 2 weeks ago
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Universal components for differentiable scientific computing.☆135Updated this week
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆15Oct 6, 2026Updated last week
- Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡☆37Updated this week
- Website for the book "The Elements of Differentiable Programming".☆15Jul 2, 2025Updated last year
- ☆15Feb 18, 2024Updated 2 years ago
- Chrome browser extension that adds customizable keyboard shortcuts to the Overleaf online latex editor.☆14Jun 22, 2024Updated 2 years ago
- Diffusion MRI Simulation Toolbox in Julia☆21Jan 22, 2023Updated 3 years ago
- Special functions in Jax☆20Updated this week
- ☆36Jul 11, 2017Updated 9 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Transformer creation from scratch using Jax.☆20Dec 8, 2024Updated last year
- Simple differentiable approximate ocean models built with JAX.☆18Oct 13, 2023Updated 3 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆626Aug 11, 2026Updated 2 months ago
- Orthogonal polynomials with JAX☆31Oct 5, 2026Updated last week
- GPU-accelerated differentiable sparse linear solver library for JAX☆57Oct 8, 2026Updated last week
- ☆53Jun 8, 2026Updated 4 months ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- Pseudospectral Kolmogorov Flow Solver☆43Oct 24, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆448Nov 14, 2025Updated 11 months ago
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆21Updated this week
- [ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch☆86Nov 21, 2025Updated 10 months ago
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆38Apr 21, 2025Updated last year
- Compute gradients on mesh and unstructured data objects.☆15Jan 30, 2021Updated 5 years ago
- ☆17Updated this week
- Generate Streamlit apps from Tesseracts, automagically.☆17Updated this week