TUM-DAML / seml
SEML: Slurm Experiment Management Library
☆166Updated last week
Related projects: ⓘ
- A Machine Learning workflow for Slurm.☆145Updated 3 years ago
- A library for programmatically generating equivariant layers through constraint solving☆252Updated last year
- Example of how to use Weights & Biases on Slurm☆108Updated 2 years ago
- ☆257Updated last year
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated 10 months ago
- Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.☆502Updated this week
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆555Updated 4 months ago
- Large-scale, multi-GPU capable, kernel solver☆179Updated 2 months ago
- Turning SymPy expressions into PyTorch modules.☆139Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆300Updated 6 months ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆124Updated 2 years ago
- ☆215Updated 2 years ago
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆207Updated 3 months ago
- Code for the paper "Topological Autoencoders" by Michael Moor, Max Horn, Bastian Rieck, and Karsten Borgwardt.☆138Updated 2 years ago
- ☆151Updated 4 years ago
- Computing on Topological Domains☆175Updated last month
- Algorithms for computations on random manifolds made easier☆82Updated 9 months ago
- Topological Graph Neural Networks (ICLR 2022)☆109Updated 2 years ago
- VAEs with Lie Group latent space☆97Updated 2 years ago
- Manifold-learning flows (ℳ-flows)☆230Updated 3 years ago
- GFlowNet library☆213Updated 3 weeks ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆106Updated this week
- pyrelational is a python active learning library for rapidly implementing active learning pipelines from data management, model developme…☆151Updated last week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆231Updated this week
- EigenPro Iteration in PyTorch☆19Updated 8 months ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆50Updated last year
- ☆120Updated last year
- COMP760 Lecture Notes☆24Updated last year
- Tools to connect to and interact with the Mila cluster☆60Updated last month
- A topological machine learning framework based on PyTorch☆147Updated 3 weeks ago