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.☆99Updated last month
- Agustinus' very opiniated publication-ready plotting library☆71Updated 9 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆124Updated 9 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Running Jax in PyTorch Lightning☆119Updated last year
- Algorithms for computations on random manifolds made easier☆94Updated 2 years ago
- JAX Arrays for human consumption☆111Updated 3 months ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆208Updated last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆152Updated 2 months ago
- Official PyTorch implementation of NeuralSVD (ICML 2024)☆22Updated last year
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆95Updated 3 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆110Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆62Updated this week
- Truncated Normal Distribution in PyTorch☆87Updated 2 years ago
- Deep memory and sequence models in JAX☆23Updated 3 weeks ago
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆252Updated last year
- Sketched linear operations for PyTorch☆100Updated 3 months 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
- Uncertainty quantification with PyTorch☆379Updated 2 weeks ago
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆727Updated 2 weeks ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆35Updated last month
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆700Updated last week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆311Updated this week
- IVON optimizer for neural networks based on variational learning.☆81Updated last year
- ☆619Updated 2 weeks ago
- Code for our paper: Online Variational Filtering and Parameter Learning☆20Updated 4 years ago
- Simple (and cheap!) neural network uncertainty estimation☆79Updated 4 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year