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.☆95Updated 5 months ago
- Stochastic Automatic Differentiation library for PyTorch.☆208Updated last year
- Running Jax in PyTorch Lightning☆113Updated 10 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆294Updated this week
- ☆40Updated 3 years ago
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Use Jax functions in Pytorch☆253Updated 2 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆145Updated 5 months ago
- IVON optimizer for neural networks based on variational learning.☆72Updated last year
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆92Updated 3 years ago
- Tools to connect to and interact with the Mila cluster☆77Updated this week
- Hierarchical Associative Memory User Experience☆104Updated 3 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆108Updated last year
- A simple library for scaling up JAX programs☆144Updated this week
- Sketched linear operations for PyTorch☆96Updated 2 weeks ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆35Updated last week
- Parameter-Free Optimizers for Pytorch☆131Updated last year
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆84Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 9 months ago
- Official PyTorch implementation of NeuralSVD (ICML 2024)☆20Updated last year
- Run PyTorch in JAX. 🤝☆305Updated 3 weeks ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆57Updated last week
- LoRA for arbitrary JAX models and functions☆142Updated last year
- Truncated Normal Distribution in PyTorch☆86Updated last year
- Agustinus' very opiniated publication-ready plotting library☆69Updated 6 months ago
- Automatic Functional Differentiation in JAX☆79Updated last month
- A collection of meta-learning algorithms in Jax☆23Updated 3 years ago
- Example of how to use Weights & Biases on Slurm☆118Updated 3 years ago
- JAX Arrays for human consumption☆109Updated 2 weeks ago