alonfnt / pcax
Minimal Principal Component Analysis (PCA) implementation in JAX
☆15Updated 6 months ago
Related projects: ⓘ
- ☆49Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆47Updated 7 months ago
- JAX Arrays for human consumption☆87Updated last year
- Normalizing Flows using JAX☆82Updated 9 months ago
- scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch☆17Updated this week
- Bayesian inference with Python and Jax.☆31Updated last year
- Pytorch-like dataloaders in JAX.☆52Updated last month
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆39Updated 11 months ago
- ☆104Updated last week
- Agustinus' very opiniated publication-ready plotting library☆51Updated 2 weeks ago
- Hierarchical Associative Memory User Experience☆88Updated 8 months ago
- Normalizing Flows with a resampled base distribution☆43Updated 2 years ago
- A functional training loops library for JAX☆84Updated 7 months ago
- Multiple dispatch over abstract array types in JAX.☆100Updated last month
- Implementation of Action Matching☆35Updated last year
- Riemannian Optimization Using JAX☆44Updated 10 months ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆34Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆32Updated 3 weeks ago
- ☆85Updated this week
- ☆30Updated 3 months ago
- ☆20Updated last year
- Stochastic Normalizing Flows☆72Updated 2 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆13Updated this week
- Signax: Signature computation in JAX☆23Updated 3 weeks ago
- Turn jitted jax functions back into python source code☆20Updated 2 months ago
- Lightning-like training API for JAX with Flax☆28Updated 4 months ago
- JMP is a Mixed Precision library for JAX.☆183Updated 3 months ago
- Bayesian algorithm execution (BAX)☆46Updated 3 years ago
- Add a tqdm progress bar to your JAX scans and loops.☆66Updated 2 months ago
- IVON optimizer for neural networks based on variational learning.☆45Updated last month