ALRhub / hydra_list_sweeperLinks
Implementation of a list sweeper: instead of the cartesian product, sweep over the zipped list
☆25Updated 8 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 last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆146Updated 3 weeks ago
- Agustinus' very opiniated publication-ready plotting library☆69Updated 7 months ago
- Pytorch-like dataloaders for JAX.☆98Updated 6 months ago
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆715Updated 5 months ago
- Uncertainty quantification with PyTorch☆376Updated 2 months ago
- Running Jax in PyTorch Lightning☆115Updated 11 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Official PyTorch implementation of NeuralSVD (ICML 2024)☆20Updated last year
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆206Updated last year
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆675Updated 2 weeks ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆94Updated 3 years ago
- ☆608Updated this week
- Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)☆21Updated 5 months ago
- Truncated Normal Distribution in PyTorch☆87Updated last year
- A framework for composing Neural Processes in Python☆89Updated 11 months ago
- Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.☆227Updated last year
- IVON optimizer for neural networks based on variational learning.☆76Updated last year
- Simple (and cheap!) neural network uncertainty estimation☆78Updated 2 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated 10 months ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆36Updated last month
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆38Updated 2 months ago
- Algorithms for computations on random manifolds made easier☆94Updated 2 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆61Updated last year
- ☆312Updated 11 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆120Updated 7 months ago
- Code for our paper: Online Variational Filtering and Parameter Learning☆20Updated 4 years ago
- JAX Arrays for human consumption☆110Updated last month
- Signax: Signature computation in JAX☆37Updated 10 months ago