vinayprabhu / X-is-all-you-need

Related projects ⓘ

Alternatives and complementary repositories for X-is-all-you-need

• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 3 years ago
• srush / torch-queue
  ☆67Updated last year
• rasbt / faster-pytorch-blog
  Outlining techniques for improving the training performance of your PyTorch model without compromising its accuracy
  ☆124Updated last year
• pietrolesci / energizer
  An active learning library for Pytorch based on Lightning-Fabric.
  ☆79Updated 6 months ago
• vpj / annotate
  Annotate python source code
  ☆69Updated 4 years ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆43Updated 4 months ago
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆237Updated last year
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆185Updated 2 years ago
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆105Updated last year
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆66Updated last year
• HomebrewNLP / HomebrewNLP
  A case study of efficient training of large language models using commodity hardware.
  ☆68Updated 2 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆55Updated 3 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆322Updated last year
• theblackcat102 / H5Record
  Large dataset storage format for Pytorch
  ☆45Updated 3 years ago
• LaihoE / did-it-spill
  Check if you have training samples in your test set
  ☆64Updated 2 years ago
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆210Updated last year
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆43Updated 3 years ago
• aleatory-science / deep-probprog-course
  Deep Probabilistic Programming Course @ DIKU
  ☆56Updated 4 years ago
• google-deepmind / jaxline
  ☆152Updated 11 months ago
• ahthie7u / cockpit
  Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"
  ☆31Updated 4 years ago
• clemkoa / ntm
  Neural Turing Machines in pytorch
  ☆47Updated 2 years ago
• nestordemeure / flaxOptimizers
  A collection of optimizers, some arcane others well known, for Flax.
  ☆29Updated 3 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆116Updated 3 months ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 3 years ago
• UmerHA / triton_util
  Make triton easier
  ☆41Updated 5 months ago
• namedtensor / notation
  ☆108Updated last year