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
- Pytorch-like dataloaders for JAX.☆98Updated last month
- Add a tqdm progress bar to your JAX scans and loops.☆124Updated 8 months ago
- Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)☆21Updated 7 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆151Updated 2 months ago
- Running Jax in PyTorch Lightning☆119Updated last year
- Truncated Normal Distribution in PyTorch☆87Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Uncertainty quantification with PyTorch☆379Updated this week
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆693Updated 3 weeks ago
- Simple (and cheap!) neural network uncertainty estimation☆79Updated 3 months ago
- Agustinus' very opiniated publication-ready plotting library☆70Updated 8 months ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆94Updated 3 years ago
- Signax: Signature computation in JAX☆37Updated last year
- Gaussian processes in JAX and Flax.☆578Updated last week
- ☆618Updated last week
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated last year
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆208Updated last year
- Algorithms for computations on random manifolds made easier☆94Updated 2 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆309Updated last week
- A modular, easy to extend GFlowNet library☆304Updated this week
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆251Updated last year
- JAX Arrays for human consumption☆110Updated 3 months ago
- ☆211Updated last week
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Deep memory and sequence models in JAX☆22Updated 2 weeks ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆310Updated 2 weeks ago
- A framework for composing Neural Processes in Python☆89Updated last year
- Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.☆227Updated last year
- Code for our paper: Online Variational Filtering and Parameter Learning☆20Updated 4 years ago