f-dangel / cockpit

Related projects ⓘ

Alternatives and complementary repositories for cockpit

• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• f-dangel / backpack
  BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
  ☆561Updated this week
• 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
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• szymonmaszke / torchlayers
  Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
  ☆569Updated 2 years ago
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆661Updated 2 years ago
• google-research / robustness_metrics
  ☆466Updated 3 months ago
• patrick-kidger / torchtyping
  Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
  ☆1,408Updated 3 months ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆389Updated last week
• 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
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆253Updated 2 years ago
• michaelsdr / momentumnet
  Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities
  ☆208Updated 6 months ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,396Updated this week
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆545Updated 3 years ago
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆695Updated last year
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆774Updated 10 months ago
• facebookresearch / bitsandbytes
  Library for 8-bit optimizers and quantization routines.
  ☆714Updated 2 years ago
• facebookresearch / ppuda
  Code for Parameter Prediction for Unseen Deep Architectures (NeurIPS 2021)
  ☆483Updated last year
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆471Updated this week
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆896Updated this week
• BlackHC / toma
  Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
  ☆428Updated 2 months ago
• suriyadeepan / torchtest
  Unit Testing for pytorch, based on mltest
  ☆311Updated 4 years ago
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆363Updated last year
• uclnlp / torch-imle
  Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions
  ☆258Updated last year
• awaelchli / pytorch-lightning-snippets
  A collection of code snippets for my PyTorch Lightning projects
  ☆107Updated 3 years ago
• lkhphuc / lightning-hydra-template
  Deep Learning project template best practices with Pytorch Lightning, Hydra, Tensorboard.
  ☆155Updated 3 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆151Updated 2 years ago
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆470Updated last year
• amirgholami / adahessian
  ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
  ☆266Updated last year