AndreasBergmeister / pjaxLinks
A projection-based framework for gradient-free and parallel learning
☆104Updated 2 months ago
Alternatives and similar repositories for pjax
Users that are interested in pjax are comparing it to the libraries listed below
Sorting:
- Run PyTorch in JAX. 🤝☆283Updated last week
- Deep Learning, an Energy Approach☆201Updated 2 months ago
- Compositional Linear Algebra☆490Updated last month
- Code for the book "The Elements of Differentiable Programming".☆260Updated 2 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆108Updated last year
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆463Updated 2 weeks ago
- Patched Attention for Nonlinear Dynamics☆156Updated last week
- JAX Arrays for human consumption☆106Updated last month
- ☆30Updated last month
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Interactive textbook on state-space models☆197Updated last year
- Loopy belief propagation for factor graphs on discrete variables in JAX☆154Updated 10 months ago
- Uncertainty quantification with PyTorch☆372Updated 4 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆116Updated 3 months ago
- Hundreds of strange attractors☆487Updated last week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆38Updated last week
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 7 months ago
- Pytorch-like dataloaders for JAX.☆94Updated 3 months ago
- Multiple dispatch over abstract array types in JAX.☆128Updated last week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆112Updated 5 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆282Updated last month
- Materials of the Nordic Probabilistic AI School 2022.☆180Updated 2 years ago
- Agustinus' very opiniated publication-ready plotting library☆69Updated 3 months ago
- Flexible Inference for Predictive Coding Networks in JAX.☆53Updated this week
- ☆209Updated 2 weeks ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆57Updated last month
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆137Updated 2 months ago
- Parallel hyperparameter tuning with JAX☆36Updated last month
- ☆115Updated 2 months ago
- Matrix-free linear algebra in JAX.☆138Updated last week