AndreasBergmeister / pjaxLinks
A projection-based framework for gradient-free and parallel learning
☆97Updated 3 weeks ago
Alternatives and similar repositories for pjax
Users that are interested in pjax are comparing it to the libraries listed below
Sorting:
- Code for the book "The Elements of Differentiable Programming".☆241Updated 3 weeks ago
- Compositional Linear Algebra☆478Updated last month
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆437Updated last month
- Deep Learning, an Energy Approach☆187Updated last month
- Patched Attention for Nonlinear Dynamics☆150Updated 2 weeks ago
- JAX Arrays for human consumption☆106Updated last week
- Run PyTorch in JAX. 🤝☆256Updated last week
- Agustinus' very opiniated publication-ready plotting library☆67Updated 2 months ago
- Uncertainty quantification with PyTorch☆362Updated 3 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆32Updated this week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 4 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆110Updated 2 months ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆154Updated 8 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆46Updated 5 months ago
- Matrix-free linear algebra in JAX.☆126Updated last month
- ☆29Updated 5 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Algorithms for computations on random manifolds made easier☆91Updated last year
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆29Updated 2 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆280Updated 2 weeks ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆269Updated this week
- A Python package of computer vision models for the Equinox ecosystem.☆107Updated 11 months ago
- ☆196Updated 3 weeks ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆129Updated last month
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆457Updated 2 weeks ago
- Gaussian processes in JAX and Flax.☆521Updated 3 weeks ago
- ☆110Updated last month
- Turn SymPy expressions into trainable JAX expressions.☆343Updated 2 months ago
- Riemannian Optimization Using JAX☆49Updated last year
- Toolbox for working with streaming data as rough paths in Python☆36Updated 2 weeks ago