f0uriest / orthaxView external linksLinks
Orthogonal polynomials with JAX
☆26Jan 20, 2026Updated 3 weeks ago
Alternatives and similar repositories for orthax
Users that are interested in orthax are comparing it to the libraries listed below
Sorting:
- Numerical quadrature with JAX☆75Jan 20, 2026Updated 3 weeks ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆115Feb 3, 2026Updated last week
- Coordinates in JAX☆29Jan 31, 2026Updated last week
- Interpolation and function approximation with JAX☆248Feb 2, 2026Updated last week
- Batched routines (BLAS, LAPACK, etc.) for multi-dimensional arrays☆12Apr 10, 2022Updated 3 years ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 5 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Updated this week
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 2 months ago
- Flexible, general-purpose VMC framework, built on JAX.☆29Nov 25, 2025Updated 2 months ago
- Repository for material accompanying the ptvmc systematic study paper☆16Nov 17, 2025Updated 2 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆102Apr 24, 2025Updated 9 months ago
- GPU-accelerated coupled cluster with density fitting☆17Jun 26, 2018Updated 7 years ago
- A collection of lectures and tutorials☆24Jun 14, 2024Updated last year
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆18Apr 20, 2023Updated 2 years ago
- A JAX-based gaussian process structure learning package for time series modelling.☆34May 28, 2025Updated 8 months ago
- Simulation-based inference in JAX☆45Nov 24, 2025Updated 2 months ago
- A differentiable parametrization of a group of unitary matrices.☆14May 13, 2021Updated 4 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Jan 16, 2026Updated 3 weeks ago
- ML-Based Quantum Error Decoder☆13Jun 10, 2018Updated 7 years ago
- 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
- 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 based on Tensorflow for running variational Monte-Carlo simulations of quantum many-body systems.☆38Oct 30, 2020Updated 5 years ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated last year
- Unitful Quantities in JAX☆52Updated this week
- MPS in Julia minicourse☆19Oct 13, 2017Updated 8 years ago
- Differentiable programming on quantum circuits with Flux☆19Feb 12, 2020Updated 6 years ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Matrix-free linear algebra in JAX.☆159Dec 2, 2025Updated 2 months ago
- Normalizing Flows with a resampled base distribution☆47Sep 7, 2022Updated 3 years ago
- ☆43Feb 6, 2026Updated last week
- Python code for the linear inverse problem☆24Sep 18, 2022Updated 3 years ago
- A JAX library for Density Functional Theory.☆54Nov 25, 2025Updated 2 months ago
- Pytorch-like dataloaders for JAX.☆99Dec 16, 2025Updated last month
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆32Apr 19, 2025Updated 9 months ago
- Convolutional Differential Operators for Physics-based Deep Learning Study☆25Jul 30, 2024Updated last year
- Parallel hyperparameter tuning with JAX☆39Jul 21, 2025Updated 6 months ago
- Differentiable optical models as parameterised neural networks in Jax using Zodiax☆64Dec 1, 2025Updated 2 months ago
- Spiking neural networks in JAX with equinox.☆28Aug 29, 2024Updated last year