vinayprabhu / X-is-all-you-need

Related projects: ⓘ

• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated 11 months ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆71Updated 2 years ago
• srush / torch-queue
  ☆64Updated 10 months ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆55Updated 3 years ago
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆66Updated last year
• facebookresearch / torchrec
  ☆40Updated this week
• theblackcat102 / H5Record
  Large dataset storage format for Pytorch
  ☆45Updated 3 years ago
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆101Updated last year
• aleatory-science / deep-probprog-course
  Deep Probabilistic Programming Course @ DIKU
  ☆56Updated 4 years ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆58Updated 2 years ago
• HomebrewNLP / HomebrewNLP
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 2 years ago
• EleutherAI / pyfra
  Python Research Framework
  ☆107Updated last year
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆115Updated last month
• ischlag / fast-weight-transformers
  Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.
  ☆97Updated 3 years ago
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆47Updated 3 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆87Updated 2 years ago
• ahthie7u / cockpit
  Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"
  ☆31Updated 3 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆41Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆36Updated 5 months ago
• jxbz / nero
  👑 Pytorch code for the Nero optimiser.
  ☆20Updated last year
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆46Updated 2 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆50Updated 3 years ago
• borisdayma / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆54Updated 2 years ago
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆57Updated last month
• Felix-Petersen / diffsort
  Differentiable Sorting Networks
  ☆103Updated last year
• LaihoE / did-it-spill
  Check if you have training samples in your test set
  ☆64Updated 2 years ago
• Zasder3 / open_clip_juwels
  An open source implementation of CLIP.
  ☆32Updated last year
• mle-infrastructure / mle-logging
  Lightweight ML Experiment Logging 📖
  ☆79Updated 3 weeks ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆103Updated 2 years ago