uclnlp / torch-imle

Related projects ⓘ

Alternatives and complementary repositories for torch-imle

• Felix-Petersen / diffsort
  Differentiable Sorting Networks
  ☆107Updated last year
• y0ast / slurm-for-ml
  A Machine Learning workflow for Slurm.
  ☆146Updated 3 years ago
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆105Updated last year
• g-benton / loss-surface-simplexes
  ☆97Updated 2 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• HEmile / storchastic
  Stochastic Automatic Differentiation library for PyTorch.
  ☆180Updated 2 months ago
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆301Updated 3 weeks ago
• f-dangel / backpack
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆561Updated this week
• srush / parallax
  ☆155Updated 4 years ago
• ischlag / fast-weight-transformers
  Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.
  ☆100Updated 3 years ago
• google-research / ott
  ☆213Updated 2 years ago
• nalexai / hyperlib
  Library that contains implementations of machine learning components in the hyperbolic space
  ☆132Updated 7 months ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated last year
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆95Updated 3 years ago
• RobertCsordas / transformer_generalization
  The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…
  ☆66Updated last year
• michaelsdr / momentumnet
  Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities
  ☆208Updated 6 months ago
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆122Updated 4 months ago
• srush / torch-queue
  ☆67Updated last year
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• unixpickle / sk2torch
  Convert scikit-learn models to PyTorch modules
  ☆157Updated 6 months ago
• SirRob1997 / Crowded-Valley---Results
  This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"
  ☆180Updated 3 years ago
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆526Updated this week
• martius-lab / blackbox-backprop
  Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…
  ☆340Updated 2 years ago
• LeviViana / torch_sampling
  Efficient reservoir sampling implementation for PyTorch
  ☆104Updated 3 years ago
• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆77Updated 4 years ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆179Updated 3 months ago
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆257Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆127Updated last month
• ctlllll / SGConv
  ☆164Updated last year
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆661Updated 2 years ago