Neural Emulator Architectures in JAX.
☆25Jun 1, 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.☆13Jun 1, 2026Updated last month
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆45Nov 21, 2025Updated 8 months ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆215Jun 1, 2026Updated last month
- Official implementation of "Guiding diffusion models to reconstruct flow fields from sparse data"☆21Jan 20, 2026Updated 6 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆107Jun 1, 2026Updated last month
- 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.
- TorchFSM: Fourier Spectral Method with PyTorch☆60Jul 15, 2026Updated last week
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆34May 30, 2025Updated last year
- P3D☆36Apr 11, 2026Updated 3 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 2 years ago
- "Neural Emulator Superiority: When Machine Learning for PDEs Surpasses its Training Data" [NeurIPS 2025]☆17May 21, 2026Updated 2 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆135May 7, 2026Updated 2 months ago
- Coordinates in JAX☆35Updated this week
- Differentiable Hydrodynamical Simulations☆17Jul 17, 2026Updated last week
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated last year
- Universal components for differentiable scientific computing.☆116Updated this week
- A GPU-accelerated and differentiable library for landscape genetics and ecological connectivity modelling.☆13Jun 4, 2026Updated last month
- Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡☆32Updated this week
- Website for the book "The Elements of Differentiable Programming".☆15Jul 2, 2025Updated last year
- ☆14Feb 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
- Examples of Automatic Differentiation (AD) in many different languages and systems☆27Jun 25, 2018Updated 8 years ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆26Jun 2, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Diffusion MRI Simulation Toolbox in Julia☆21Jan 22, 2023Updated 3 years ago
- Special functions in Jax☆18Mar 1, 2025Updated last year
- ☆36Jul 11, 2017Updated 9 years ago
- 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 2 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆604May 13, 2026Updated 2 months ago
- Orthogonal polynomials with JAX☆29Updated this week
- [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method☆117Jun 1, 2026Updated last month
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- GPU-accelerated differentiable sparse linear solver library for JAX☆35Updated this week
- 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
- ☆440Nov 14, 2025Updated 8 months ago
- Spectral / Galerkin / PINNs experimentation toolkit built on top of JAX for fast differentiable ODE / PDE prototyping, variational forms,…☆20Jun 24, 2026Updated last month
- [ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch☆82Nov 21, 2025Updated 8 months ago
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆38Apr 21, 2025Updated last year