Orthogonal polynomials with JAX
☆32Oct 30, 2026Updated this week
Alternatives and similar repositories for orthax
Users that are interested in orthax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Numerical quadrature with JAX☆92Updated this week
- Coordinates in JAX☆38Oct 26, 2026Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆144Oct 21, 2026Updated last week
- Fast, lightweight and parallelised simulation-based inference in JAX.☆24Oct 28, 2025Updated last year
- JAX-compatible, simplex method-based linear program solver☆11May 24, 2026Updated 5 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Unitful Quantities in JAX☆70Updated this week
- Simulation-based inference in JAX☆55Oct 14, 2026Updated 2 weeks ago
- Interpolation and function approximation with JAX☆283Updated this week
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated last year
- Bijections & normalizing flows with JAX/NNX☆30Aug 23, 2026Updated 2 months ago
- Fully and Partially Bayesian Neural Nets☆84Apr 30, 2025Updated last year
- ☆246Oct 23, 2026Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆83Feb 12, 2026Updated 8 months ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Object-oriented Jax framework extending Equinox for scientific programming☆28Sep 23, 2026Updated last month
- Matrix-free linear algebra in JAX.☆183Oct 22, 2026Updated last week
- Fast Kolmogorov-Arnold Network in JAX, initial experiments☆16May 20, 2024Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆136May 7, 2026Updated 5 months ago
- Astronomy Research + JAX Meeting 2024☆13Oct 23, 2024Updated 2 years ago
- SParse lINear solvers in JAX☆59Aug 4, 2026Updated 2 months ago
- Neural Emulator Architectures in JAX.☆28Aug 25, 2026Updated 2 months ago
- Curated list of astronomy/astrophysics code packages using JAX☆61Mar 25, 2026Updated 7 months ago
- Solar system orbit fitting and integration with JAX☆17Updated this week
- 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.
- Pure JAX implementation of the Non-Uniform Fast Fourier Transform (NUFFT)☆29Oct 1, 2026Updated last month
- Add a tqdm progress bar to your JAX scans and loops.☆131Updated this week
- A JAX-based gaussian process structure learning package for time series modelling.☆37Apr 28, 2026Updated 6 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆110Aug 8, 2026Updated 2 months ago
- Finite difference tools in JAX☆23Feb 4, 2024Updated 2 years ago
- Special functions in Jax☆20Updated this week
- A JAX implementation of stochastic addition.☆13Aug 15, 2022Updated 4 years ago
- Differentiable Physical Optics Propagation in Jax☆80Oct 26, 2026Updated last week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆332Oct 13, 2026Updated 3 weeks 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.
- AMI-GO brrrrrrr☆26Oct 15, 2026Updated 2 weeks ago
- Generative Models for Simulation-Based Inference in JAX☆21Oct 27, 2026Updated last week
- JAXMg: A multi-GPU linear solver in JAX☆113Oct 22, 2026Updated last week
- A project to illustrate the power of Jax for simulating astrochemical kinetics.☆11Oct 12, 2026Updated 3 weeks ago
- A multidimensional spline interpolation by Google/JAX☆20Feb 13, 2024Updated 2 years ago
- An Automized Partial Differential Equation solver based on JAX☆14Jul 9, 2026Updated 3 months ago
- Persistent Sampling for fast Bayesian posterior sampling and model evidence estimation in scientific applications☆14Jun 5, 2026Updated 4 months ago