CSCfi / slurm-hyper-searchLinks
☆14Updated 4 years ago
Alternatives and similar repositories for slurm-hyper-search
Users that are interested in slurm-hyper-search are comparing it to the libraries listed below
Sorting:
- Pytorch-like dataloaders for JAX.☆85Updated last week
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆39Updated last month
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- ☆34Updated 2 years ago
- Use Jax functions in Pytorch☆242Updated last year
- ☆52Updated 2 years ago
- Riemannian Optimization Using JAX☆49Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆33Updated last year
- Parameter-Free Optimizers for Pytorch☆129Updated last year
- IVON optimizer for neural networks based on variational learning.☆66Updated 7 months ago
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆22Updated 3 years ago
- ☆114Updated this week
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆101Updated last year
- Code for the paper: "Independent mechanism analysis, a new concept?"☆24Updated last year
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- Automatic Functional Differentiation in JAX☆72Updated 2 months ago
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- Normalizing Flows using JAX☆83Updated last year
- Implementation of Action Matching☆44Updated 2 years ago
- Pytorch implementation of the Power Spherical distribution☆74Updated 10 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆34Updated 5 months ago
- ☆34Updated 2 months ago
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆15Updated 4 months ago
- Example of how to use Weights & Biases on Slurm☆115Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆45Updated 4 months ago
- JAX Arrays for human consumption☆91Updated last year
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Maximal Update Parametrization (μP) with Flax & Optax.☆11Updated last year