ALRhub / hydra_list_sweeperLinks
Implementation of a list sweeper: instead of the cartesian product, sweep over the zipped list
☆28Updated 10 months ago
Alternatives and similar repositories for hydra_list_sweeper
Users that are interested in hydra_list_sweeper are comparing it to the libraries listed below
Sorting:
- WandB sweeps integration with Hydra sweeper☆50Updated 2 years ago
- Agustinus' very opiniated publication-ready plotting library☆71Updated 9 months ago
- Pytorch-like dataloaders for JAX.☆99Updated last month
- Sketched linear operations for PyTorch☆100Updated 3 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆152Updated 2 months ago
- Uncertainty quantification with PyTorch☆379Updated 2 weeks ago
- Running Jax in PyTorch Lightning☆119Updated last year
- Code for our paper: Online Variational Filtering and Parameter Learning☆20Updated 4 years ago
- Signax: Signature computation in JAX☆38Updated last year
- A framework for composing Neural Processes in Python☆89Updated last year
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Add a tqdm progress bar to your JAX scans and loops.☆124Updated 9 months ago
- ☆622Updated 2 weeks ago
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆252Updated last year
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆700Updated last week
- JAX Arrays for human consumption☆111Updated 3 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆311Updated this week
- ☆216Updated last week
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆35Updated last month
- Algorithms for computations on random manifolds made easier☆94Updated 2 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆310Updated last month
- Truncated Normal Distribution in PyTorch☆87Updated 2 years ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆95Updated 3 years ago
- Normalizing flows in PyTorch☆442Updated 2 months ago
- Simple (and cheap!) neural network uncertainty estimation☆79Updated 4 months ago
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Updated 2 years ago
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆73Updated 9 months ago
- Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.☆228Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆61Updated 5 months ago