ALRhub / hydra_list_sweeper
Implementation of a list sweeper: instead of the cartesian product, sweep over the zipped list
☆22Updated 2 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
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆53Updated 6 months ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆69Updated 3 weeks ago
- Pytorch-like dataloaders for JAX.☆82Updated 2 weeks ago
- Algorithms for computations on random manifolds made easier☆89Updated last year
- Predictive Coding JAX-based library☆57Updated 3 weeks ago
- A dataloader, but for JAX☆21Updated last year
- JAX Arrays for human consumption☆92Updated last year
- Running Jax in PyTorch Lightning☆100Updated 5 months ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆30Updated 2 weeks ago
- Agustinus' very opiniated publication-ready plotting library☆64Updated last week
- diffusionjax is a simple and accessible diffusion models package in JAX☆42Updated 3 months ago
- Hierarchical Associative Memory User Experience☆101Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆100Updated last year
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆30Updated last month
- Stainless neural networks in JAX☆34Updated last month
- Parameter-Free Optimizers for Pytorch☆127Updated last year
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆59Updated 11 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 9 months ago
- This repository contains code for applying Riemannian geometry in machine learning.☆77Updated 3 years ago
- Code for the paper: "Independent mechanism analysis, a new concept?"☆24Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆119Updated last month
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆87Updated 3 years ago
- Official code for the ICLR 2021 paper Neural ODE Processes☆72Updated 3 years ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆38Updated 3 weeks ago
- ☆52Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆42Updated last year
- ☆114Updated this week