AndreasBergmeister / pjaxLinks
A projection-based framework for gradient-free and parallel learning
☆104Updated 3 months ago
Alternatives and similar repositories for pjax
Users that are interested in pjax are comparing it to the libraries listed below
Sorting:
- Deep Learning, an Energy Approach☆210Updated 3 months ago
- Code for the book "The Elements of Differentiable Programming".☆266Updated 3 months ago
- Patched Attention for Nonlinear Dynamics☆156Updated this week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆468Updated 2 weeks ago
- Run PyTorch in JAX. 🤝☆288Updated 2 weeks ago
- Compositional Linear Algebra☆490Updated last month
- JAX Arrays for human consumption☆106Updated 2 months ago
- Uncertainty quantification with PyTorch☆372Updated 5 months ago
- Hundreds of strange attractors☆489Updated last week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆38Updated last month
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 8 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆117Updated 4 months ago
- The boundary of neural network trainability is fractal☆216Updated last year
- Loopy belief propagation for factor graphs on discrete variables in JAX☆156Updated 11 months ago
- Interactive textbook on state-space models☆197Updated last year
- Minimal, lightweight JAX implementations of popular models.☆108Updated this week
- Agustinus' very opiniated publication-ready plotting library☆69Updated 4 months ago
- Pytorch-like dataloaders for JAX.☆94Updated 3 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆283Updated this week
- ☆119Updated 3 months ago
- Flexible Inference for Predictive Coding Networks in JAX.☆53Updated this week
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆215Updated last year
- Turn SymPy expressions into trainable JAX expressions.☆349Updated 4 months ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆472Updated 3 weeks ago
- Flow-matching algorithms in JAX☆105Updated last year
- Parallel hyperparameter tuning with JAX☆36Updated 2 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆56Updated last month
- Oscillatory State-Space Models☆103Updated 6 months ago
- Matrix-free linear algebra in JAX.☆140Updated 3 weeks ago