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.☆99Updated last month
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Updated last year
- ☆42Updated 3 years ago
- Sketched linear operations for PyTorch☆100Updated 3 months ago
- IVON optimizer for neural networks based on variational learning.☆81Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆109Updated last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆152Updated 2 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆64Updated 2 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆208Updated last year
- Official PyTorch implementation of NeuralSVD (ICML 2024)☆22Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆62Updated this week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆311Updated this week
- Transformers with doubly stochastic attention☆53Updated 3 years ago
- Hierarchical Associative Memory User Experience☆106Updated 3 weeks ago
- JAX Arrays for human consumption☆111Updated 3 months ago
- Agustinus' very opiniated publication-ready plotting library☆71Updated 9 months ago
- Code for the paper: "Independent mechanism analysis, a new concept?"☆25Updated 2 years ago
- ☆52Updated 2 years ago
- Riemannian Optimization Using JAX☆54Updated 2 years ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 11 months ago
- [ICML2022] Variational Wasserstein gradient flow☆24Updated 3 years ago
- Running Jax in PyTorch Lightning☆119Updated last year
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆46Updated 2 years ago
- Hessian spectral density estimation in TF and Jax☆125Updated 5 years ago
- Code for our paper: Online Variational Filtering and Parameter Learning☆20Updated 4 years ago
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆23Updated 5 years ago
- A simple library for scaling up JAX programs☆145Updated 3 months ago