Alternatives and similar repositories for torch-imle:

Users that are interested in torch-imle are comparing it to the libraries listed below

• g-benton / loss-surface-simplexes
  ☆98Updated 3 years ago
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆306Updated 2 months ago
• Felix-Petersen / diffsort
  Differentiable Sorting Networks
  ☆110Updated last year
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• HEmile / storchastic
  Stochastic Automatic Differentiation library for PyTorch.
  ☆197Updated 6 months ago
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆112Updated last year
• srush / parallax
  ☆153Updated 4 years ago
• f-dangel / backpack
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆572Updated 2 months ago
• y0ast / slurm-for-ml
  A Machine Learning workflow for Slurm.
  ☆149Updated 4 years ago
• nec-research / tf-imle
  Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation
  ☆67Updated 2 years ago
• izmailovpavel / understandingbdl
  ☆239Updated 2 years ago
• michaelsdr / momentumnet
  Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities
  ☆209Updated 10 months ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆107Updated 2 years ago
• srush / torch-queue
  ☆68Updated last year
• ischlag / fast-weight-transformers
  Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.
  ☆102Updated 3 years ago
• mfinzi / LieConv
  ☆262Updated last year
• unixpickle / sk2torch
  Convert scikit-learn models to PyTorch modules
  ☆160Updated 9 months ago
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆177Updated 3 weeks ago
• martius-lab / blackbox-backprop
  Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…
  ☆343Updated 3 years ago
• IssamLaradji / sps
  Official code for the Stochastic Polyak step-size optimizer
  ☆138Updated 8 months ago
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆123Updated 7 months ago
• nalexai / hyperlib
  Library that contains implementations of machine learning components in the hyperbolic space
  ☆132Updated 10 months ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆667Updated 3 years ago
• cybertronai / pytorch-sso
  PyTorch-SSO: Scalable Second-Order methods in PyTorch
  ☆145Updated last year
• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆77Updated 4 years ago
• google-research / ott
  ☆213Updated 3 years ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆462Updated 3 weeks ago
• google-deepmind / jaxline
  ☆153Updated last year
• ott-jax / ott
  Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
  ☆567Updated this week
• CUAI / Neural-Manifold-Ordinary-Differential-Equations
  [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)
  ☆115Updated last year