absorensen / numbers_go_brrrrLinks
They do indeed go brrrr
☆30Updated 4 months ago
Alternatives and similar repositories for numbers_go_brrrr
Users that are interested in numbers_go_brrrr are comparing it to the libraries listed below
Sorting:
- Compositional Linear Algebra☆491Updated 2 months ago
- A projection-based framework for gradient-free and parallel learning☆106Updated 3 months ago
- Uncertainty quantification with PyTorch☆373Updated this week
- ☆189Updated 2 months ago
- Deep Learning, an Energy Approach☆215Updated 4 months ago
- ☆222Updated this week
- Minimal Implementation of Bayesian Optimization in JAX☆97Updated 5 months ago
- Pytorch-like dataloaders for JAX.☆93Updated 4 months ago
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆653Updated this week
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆218Updated last year
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆481Updated last month
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆474Updated last week
- Agustinus' very opiniated publication-ready plotting library☆69Updated 5 months ago
- Code for the book "The Elements of Differentiable Programming".☆269Updated 3 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆277Updated this week
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆447Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆128Updated 3 weeks ago
- A Python package of computer vision models for the Equinox ecosystem.☆108Updated last year
- Gaussian processes in JAX and Flax.☆540Updated this week
- Add a tqdm progress bar to your JAX scans and loops.☆117Updated 5 months ago
- Neural Networks for JAX☆84Updated last year
- Lightning-like training API for JAX with Flax☆42Updated 10 months ago
- Spyx: Spiking Neural Networks in JAX☆125Updated 11 months ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆271Updated this week
- Machine Learning with Symbolic Tensors☆344Updated 4 months ago
- State of the art inference for your bayesian models.☆222Updated 5 months ago
- Run PyTorch in JAX. 🤝☆298Updated last month
- ☆454Updated 11 months ago
- The tiniest of Gaussian Process libraries☆318Updated last week
- Minimal, lightweight JAX implementations of popular models.☆110Updated last week