CSCfi / slurm-hyper-search
☆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:
- ☆52Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆42Updated 3 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆119Updated last month
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆38Updated 3 weeks ago
- Normalizing Flows using JAX☆83Updated last year
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Pytorch-like dataloaders for JAX.☆82Updated 2 weeks ago
- Hierarchical Associative Memory User Experience☆101Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- IVON optimizer for neural networks based on variational learning.☆65Updated 6 months ago
- JAX Arrays for human consumption☆92Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆100Updated last year
- Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)☆16Updated 3 weeks ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆272Updated this week
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Algorithms for computations on random manifolds made easier☆89Updated last year
- Gaussian-Bernoulli Restricted Boltzmann Machines☆104Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆127Updated last year
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- Multiple dispatch over abstract array types in JAX.☆121Updated 2 weeks ago
- Riemannian Optimization Using JAX☆49Updated last year
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆58Updated last year
- Graphically structured diffusion model.☆20Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆114Updated this week
- Code for the paper: "Independent mechanism analysis, a new concept?"☆24Updated last year
- Running Jax in PyTorch Lightning☆100Updated 5 months ago
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Sketched matrix decompositions for PyTorch☆70Updated 3 weeks ago
- Turn jitted jax functions back into python source code☆22Updated 5 months ago