ALRhub / hydra_list_sweeperLinks
Implementation of a list sweeper: instead of the cartesian product, sweep over the zipped list
☆23Updated 3 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☆49Updated last year
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆31Updated last month
- Pytorch-like dataloaders for JAX.☆87Updated last month
- JAX Arrays for human consumption☆93Updated last week
- Algorithms for computations on random manifolds made easier☆90Updated last year
- Official implementation of Transformer Neural Processes☆78Updated 2 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆31Updated 2 months ago
- A dataloader, but for JAX☆21Updated last year
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆87Updated 3 years ago
- ☆34Updated 2 months ago
- This repository contains code for applying Riemannian geometry in machine learning.☆77Updated 4 years ago
- Research Project Template Repository☆32Updated 2 weeks ago
- Comparison between GFlowNets & Maximum Entropy RL☆18Updated last year
- Simple (and cheap!) neural network uncertainty estimation☆66Updated last month
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆40Updated 2 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆35Updated 6 months ago
- Neural Diffusion Processes☆81Updated 10 months ago
- ☆16Updated 9 months ago
- Agustinus' very opiniated publication-ready plotting library☆66Updated last month
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆53Updated 7 months ago
- Stainless neural networks in JAX☆34Updated 2 weeks ago
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆82Updated 3 weeks ago
- Running Jax in PyTorch Lightning☆102Updated 6 months ago
- ☆32Updated last year
- ☆17Updated last year
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆55Updated last year
- Normalizing Flows with a resampled base distribution☆47Updated 2 years ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆83Updated 2 years ago
- Code for the paper: "Independent mechanism analysis, a new concept?"☆24Updated 2 years ago
- Easy Hypernetworks in Pytorch and Jax☆101Updated 2 years ago