Orthogonal polynomials with JAX
☆28Jun 22, 2026Updated last 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☆85Jun 22, 2026Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆135Jun 26, 2026Updated last week
- Coordinates in JAX☆35Updated this week
- Simulation-based inference in JAX☆52Mar 25, 2026Updated 3 months ago
- Interpolation and function approximation with JAX☆275Jun 22, 2026Updated last 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.
- Batched routines (BLAS, LAPACK, etc.) for multi-dimensional arrays☆12Apr 10, 2022Updated 4 years ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 10 months ago
- Implementation of Forward Laplacian algorithm in JAX☆100Jun 23, 2026Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆81Feb 12, 2026Updated 4 months ago
- Flexible, general-purpose VMC framework, built on JAX.☆32Jun 13, 2026Updated 3 weeks ago
- BONNI: Bayesian Optimization via Neural Network surrogates and Interior Point Optimization☆34Jun 19, 2026Updated 2 weeks ago
- Minimal Implementation of Bayesian Optimization in JAX☆108Apr 24, 2025Updated last year
- A JAX-based gaussian process structure learning package for time series modelling.☆37Apr 28, 2026Updated 2 months ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆235Jun 21, 2026Updated last week
- ☆20Apr 19, 2026Updated 2 months ago
- Fully and Partially Bayesian Neural Nets☆84Apr 30, 2025Updated last year
- Official project page for Estimating the Rate-Distortion Function by Wasserstein Gradient Descent☆19Nov 2, 2023Updated 2 years ago
- A differentiable parametrization of a group of unitary matrices.☆14May 13, 2021Updated 5 years ago
- Python code for the linear inverse problem☆24Sep 18, 2022Updated 3 years ago
- Unitful Quantities in JAX☆64Updated this week
- Fractional White Noises for Neural Stochastic Differential Equations (NeurIPS 2022)☆16Nov 17, 2022Updated 3 years ago
- Finite difference tools in JAX☆23Feb 4, 2024Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Matrix-free linear algebra in JAX.☆174Jun 23, 2026Updated last week
- dic software☆14Jun 16, 2025Updated last year
- Code for "Learning Unitary Operators with Help From u(n)", AAAI-17. (https://arxiv.org/abs/1607.04903)☆17Jan 10, 2017Updated 9 years ago
- A framework for applied category theory in the Julia language☆20Oct 11, 2024Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated 2 years ago
- Differentiable programming on quantum circuits with Flux☆19Feb 12, 2020Updated 6 years ago
- Differentiable Physical Optics Propagation in Jax☆73Jun 4, 2026Updated last month
- Repository for material accompanying the ptvmc systematic study paper☆16Nov 17, 2025Updated 7 months ago
- A collection of lectures and tutorials☆25Jun 4, 2026Updated last month
- 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.
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆47Jun 15, 2026Updated 2 weeks ago
- MPS in Julia minicourse☆19Oct 13, 2017Updated 8 years ago
- A framework based on Tensorflow for running variational Monte-Carlo simulations of quantum many-body systems.☆37Oct 30, 2020Updated 5 years ago
- Pytorch-like dataloaders for JAX.☆104Dec 16, 2025Updated 6 months ago
- SParse lINear solvers in JAX☆46Feb 22, 2026Updated 4 months ago
- A Z-order (Morton-code) like coordinate system as template library for arbitrary dimensions.☆12Nov 24, 2019Updated 6 years ago
- Cooperative workspace driver☆10Jan 20, 2026Updated 5 months ago