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 last week
- Pytorch-like dataloaders for JAX.☆96Updated 5 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆118Updated 6 months ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆35Updated 3 weeks ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆205Updated last year
- Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.☆226Updated last year
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆669Updated 2 weeks ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆309Updated 4 months ago
- Algorithms for computations on random manifolds made easier☆94Updated last year
- Truncated Normal Distribution in PyTorch☆87Updated last year
- JAX Arrays for human consumption☆110Updated last month
- ☆605Updated 3 weeks ago
- Running Jax in PyTorch Lightning☆114Updated 11 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 10 months ago
- A framework for composing Neural Processes in Python☆88Updated 11 months ago
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆93Updated 3 years ago
- Reduce multiple PyTorch TensorBoard runs to new event (or CSV) files.☆76Updated last week
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Updated 2 years ago
- Uncertainty quantification with PyTorch☆376Updated last month
- ☆312Updated 8 months ago
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆239Updated last year
- Mixture density network implemented in PyTorch.☆152Updated 2 years ago
- Simple (and cheap!) neural network uncertainty estimation☆77Updated last month
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆38Updated last month
- Signax: Signature computation in JAX☆37Updated 9 months ago
- Stainless neural networks in JAX☆34Updated 2 months ago
- CausalPFN: Amortized Causal Effect Estimation via In-Context Learning☆63Updated 2 weeks ago
- Example of how to use Weights & Biases on Slurm☆118Updated 3 years ago