f-dangel / cockpit

Related projects: ⓘ

• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• BlackHC / toma
  Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
  ☆419Updated 3 weeks ago
• google-research / robustness_metrics
  ☆462Updated last month
• f-dangel / backpack
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆555Updated 4 months ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• facebookincubator / flowtorch
  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
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆379Updated 2 months ago
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆879Updated last month
• szymonmaszke / torchlayers
  Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
  ☆568Updated 2 years ago
• suriyadeepan / torchtest
  Unit Testing for pytorch, based on mltest
  ☆309Updated 3 years ago
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆546Updated 3 years ago
• facebookresearch / ppuda
  Code for Parameter Prediction for Unseen Deep Architectures (NeurIPS 2021)
  ☆484Updated last year
• SirRob1997 / Crowded-Valley---Results
  This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"
  ☆176Updated 3 years ago
• patrick-kidger / torchtyping
  Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
  ☆1,382Updated last month
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆252Updated last year
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,389Updated this week
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆360Updated last year
• brohrer / sharpened-cosine-similarity
  An alternative to convolution in neural networks
  ☆248Updated 5 months ago
• lkhphuc / lightning-hydra-template
  Deep Learning project template best practices with Pytorch Lightning, Hydra, Tensorboard.
  ☆153Updated 3 years ago
• facebookresearch / recipes
  Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…
  ☆286Updated this week
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆767Updated 8 months ago
• decile-team / cords
  Reduce end to end training time from days to hours (or hours to minutes), and energy requirements/costs by an order of magnitude using co…
  ☆316Updated last year
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆649Updated 2 years ago
• lessw2020 / Best-Deep-Learning-Optimizers
  Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable
  ☆212Updated 3 years ago
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆447Updated this week
• leopard-ai / betty
  Betty: an automatic differentiation library for generalized meta-learning and multilevel optimization
  ☆330Updated 2 months ago
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆695Updated last year
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆211Updated 10 months ago
• uclnlp / torch-imle
  Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions
  ☆257Updated 10 months ago