TUM-DAML / seml
SEML: Slurm Experiment Management Library
☆179Updated 4 months ago
Alternatives and similar repositories for seml:
Users that are interested in seml are comparing it to the libraries listed below
- A library for programmatically generating equivariant layers through constraint solving☆265Updated last year
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- Use Jax functions in Pytorch☆238Updated last year
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆571Updated last week
- Example of how to use Weights & Biases on Slurm☆113Updated 2 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆305Updated last week
- ☆265Updated last year
- Large-scale, multi-GPU capable, kernel solver☆186Updated 8 months ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- This repository contains code for applying Riemannian geometry in machine learning.☆78Updated 3 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆572Updated 2 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆264Updated this week
- Official repository for "Categorical Normalizing Flows via Continuous Transformations"☆56Updated 4 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆112Updated this week
- Algorithms for computations on random manifolds made easier☆88Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- Compositional Linear Algebra☆465Updated last week
- Integrate pytorch☆46Updated 3 years ago
- Turning SymPy expressions into PyTorch modules.☆146Updated last year
- ☆122Updated 2 years ago
- WandB sweeps integration with Hydra sweeper☆47Updated last year
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- VAEs with Lie Group latent space☆97Updated 3 years ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆129Updated 3 years ago
- Manifold-learning flows (ℳ-flows)☆229Updated 4 years ago
- Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"☆286Updated 4 years ago
- ☆213Updated 3 years ago
- Discrete Normalizing Flows implemented in PyTorch☆110Updated 3 years ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆84Updated 2 years ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆30Updated last year