Orthogonal polynomials with JAX
☆28Jan 20, 2026Updated last month
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☆77Feb 16, 2026Updated 2 weeks ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆121Updated this week
- Coordinates in JAX☆29Jan 31, 2026Updated last month
- Interpolation and function approximation with JAX☆250Feb 16, 2026Updated 2 weeks ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 6 months ago
- Batched routines (BLAS, LAPACK, etc.) for multi-dimensional arrays☆12Apr 10, 2022Updated 3 years ago
- Implementation of Forward Laplacian algorithm in JAX☆94Nov 25, 2025Updated 3 months ago
- Flexible, general-purpose VMC framework, built on JAX.☆29Nov 25, 2025Updated 3 months ago
- JAX implementation of Kolmogorov Arnold Networks (KANs).☆10May 7, 2024Updated last year
- Repository for material accompanying the ptvmc systematic study paper☆16Nov 17, 2025Updated 3 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆104Apr 24, 2025Updated 10 months 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
- GPU-accelerated coupled cluster with density fitting☆18Jun 26, 2018Updated 7 years ago
- Simulation-based inference in JAX☆45Nov 24, 2025Updated 3 months ago
- ☆219Feb 21, 2026Updated last week
- ML-Based Quantum Error Decoder☆13Jun 10, 2018Updated 7 years 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 last month
- A JAX-based gaussian process structure learning package for time series modelling.☆36May 28, 2025Updated 9 months ago
- Fractional White Noises for Neural Stochastic Differential Equations (NeurIPS 2022)☆16Nov 17, 2022Updated 3 years ago
- Fully and Partially Bayesian Neural Nets☆80Apr 30, 2025Updated 10 months 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
- Finite difference tools in JAX☆23Feb 4, 2024Updated 2 years ago
- A framework based on Tensorflow for running variational Monte-Carlo simulations of quantum many-body systems.☆39Oct 30, 2020Updated 5 years ago
- Unitful Quantities in JAX☆53Updated this week
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Differentiable programming on quantum circuits with Flux☆19Feb 12, 2020Updated 6 years ago
- MPS in Julia minicourse☆19Oct 13, 2017Updated 8 years ago
- Matrix-free linear algebra in JAX.☆160Dec 2, 2025Updated 3 months ago
- Normalizing Flows with a resampled base distribution☆48Sep 7, 2022Updated 3 years ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Mar 22, 2024Updated last year
- Python code for the linear inverse problem☆24Sep 18, 2022Updated 3 years ago
- ☆46Feb 13, 2026Updated 3 weeks ago
- A JAX library for Density Functional Theory.☆54Nov 25, 2025Updated 3 months ago
- Pytorch-like dataloaders for JAX.☆101Dec 16, 2025Updated 2 months ago
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆34Apr 19, 2025Updated 10 months ago
- Parallel hyperparameter tuning with JAX☆39Jul 21, 2025Updated 7 months ago
- Differentiable optical models as parameterised neural networks in Jax using Zodiax☆65Feb 23, 2026Updated last week