Differentiable Principal Component Analysis (PCA) implementation in JAX
☆34Apr 19, 2025Updated 10 months ago
Alternatives and similar repositories for pcax
Users that are interested in pcax are comparing it to the libraries listed below
Sorting:
- ☆15Feb 25, 2026Updated last week
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- A minimal JAX library for connectivity modelling at scale☆11Jan 27, 2026Updated last month
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆29May 30, 2025Updated 9 months ago
- Stochastic trace estimation using JAX☆17Aug 20, 2025Updated 6 months ago
- ☆14Feb 18, 2024Updated 2 years ago
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated 8 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Jan 16, 2026Updated last month
- Add a tqdm progress bar to your JAX scans and loops.☆124May 9, 2025Updated 9 months ago
- Neural Emulator Architectures in JAX.☆23Feb 20, 2026Updated 2 weeks ago
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated last year
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Oct 12, 2023Updated 2 years ago
- Connect to Slurm Compute Nodes Directly from VSCode. Do your debugging and development easily.☆28Oct 8, 2024Updated last year
- ☆36Jul 11, 2017Updated 8 years ago
- Orthogonal polynomials with JAX☆28Jan 20, 2026Updated last month
- PROSE: Predicting Multiple Operators and Symbolic Expressions☆32Nov 10, 2025Updated 3 months ago
- ☆19Sep 22, 2025Updated 5 months ago
- Synthetic Lagrangian Turbulence by Generative Diffusion Models☆27Nov 14, 2024Updated last year
- PowerBiMIP is an open-source, efficient bilevel mixed-integer programming (BiMIP) solver, with a special focus on applications in power a…☆34Feb 26, 2026Updated last week
- Efficient Differentiable n-d PDE Solvers in JAX.☆155Updated this week
- A collection of graph neural networks implementations in JAX☆35Nov 28, 2023Updated 2 years ago
- Matrix-free linear algebra in JAX.☆160Dec 2, 2025Updated 3 months ago
- Probabilistic Programming and Nested sampling in JAX☆223Feb 27, 2026Updated last week
- Spectral Neural Operator☆79Dec 20, 2023Updated 2 years ago
- ☆19Nov 20, 2025Updated 3 months ago
- ☆10Apr 5, 2024Updated last year
- QuESt Planning is a long-term power system capacity expansion planning model that identifies cost-optimal energy storage, generation, and…☆14Feb 4, 2026Updated last month
- ☆41Apr 27, 2022Updated 3 years ago
- Code release for "Stochastic Optimal Control Matching"☆39Aug 14, 2024Updated last year
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 2 years ago
- Visualize linear programming at https://lpviz.net☆33Jan 20, 2026Updated last month
- Implementations of the renormalization group-based diffusion model (RGDM).☆16Mar 10, 2025Updated 11 months ago
- Simulation of instrumental systematic effects in the context of CMB observations☆10Apr 29, 2025Updated 10 months ago
- ☆12Mar 15, 2023Updated 2 years ago
- Example Systems using PowerDynamics.jl☆12Oct 10, 2022Updated 3 years ago
- ☆15Jan 9, 2025Updated last year
- Latest version of IFISS software☆11Dec 27, 2025Updated 2 months ago
- The official Pytorch implementation of the paper "Fourier Transformer: Fast Long Range Modeling by Removing Sequence Redundancy with FFT …☆40Mar 7, 2024Updated last year