AndreasBergmeister / pjaxLinks
A projection-based framework for gradient-free and parallel learning
☆109Updated 6 months 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".☆279Updated 6 months ago
- Deep Learning, an Energy Approach☆227Updated 6 months ago
- Patched Attention for Nonlinear Dynamics☆162Updated 2 weeks ago
- Run PyTorch in JAX. 🤝☆310Updated 2 months ago
- Compositional Linear Algebra☆504Updated 5 months ago
- Hundreds of strange attractors☆511Updated last month
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- ☆122Updated 6 months ago
- The boundary of neural network trainability is fractal☆221Updated last year
- Uncertainty quantification with PyTorch☆379Updated 2 months ago
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆41Updated 7 months ago
- Sketched linear operations for PyTorch☆100Updated 2 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆520Updated last week
- Oscillatory State-Space Models☆112Updated 2 months ago
- Interactive textbook on state-space models☆200Updated last year
- Agustinus' very opiniated publication-ready plotting library☆70Updated 7 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆150Updated last month
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated 11 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- Flow-matching algorithms in JAX☆112Updated last year
- Loopy belief propagation for factor graphs on discrete variables in JAX☆165Updated last year
- Pre-trained Gaussian processes for Bayesian optimization☆99Updated 8 months ago
- Minimal, lightweight JAX implementations of popular models.☆175Updated last week
- The Official PyTorch Implementation of "Brain-like Variational Inference" (NeurIPS 2025 Paper)☆64Updated last month
- Hierarchical Associative Memory User Experience☆105Updated last month
- Pytorch-like dataloaders for JAX.☆97Updated 2 weeks ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆106Updated 2 months ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆174Updated 2 years ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆291Updated last week
- JAX Arrays for human consumption☆110Updated 2 months ago