uclnlp / torch-imle
Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions
☆257Updated 10 months ago
Related projects: ⓘ
- Differentiable Sorting Networks☆103Updated last year
- Named tensors with first-class dimensions for PyTorch☆321Updated 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
- Differentiable Algorithms and Algorithmic Supervision.☆101Updated last year
- ☆96Updated 2 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆555Updated 4 months ago
- Stochastic Automatic Differentiation library for PyTorch.☆178Updated 2 weeks ago
- A Machine Learning workflow for Slurm.☆145Updated 3 years ago
- ☆156Updated 4 years ago
- Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.☆499Updated this week
- Laplace approximations for Deep Learning.☆447Updated this week
- Convert scikit-learn models to PyTorch modules☆157Updated 4 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆173Updated 2 years ago
- ☆215Updated 2 years ago
- Library that contains implementations of machine learning components in the hyperbolic space☆127Updated 5 months ago
- ☆290Updated 2 months ago
- Code for our NeurIPS 2022 paper☆360Updated last year
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆97Updated 3 years ago
- ☆231Updated last year
- ☆182Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆105Updated this week
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆106Updated this week
- Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…☆334Updated 2 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆176Updated last month
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆68Updated 2 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆474Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆105Updated 4 months ago
- Hyperbolic Hierarchical Clustering.☆190Updated 11 months ago
- Official codebase for Pretrained Transformers as Universal Computation Engines.☆244Updated 2 years ago