Alternatives and similar repositories for litData

Users that are interested in litData are comparing it to the libraries listed below

• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆269Updated this week
• Lightning-AI / lightning-thunder
  Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …
  ☆1,357Updated this week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆925Updated last week
• pytorch / torchft
  PyTorch per step fault tolerance (actively under development)
  ☆302Updated this week
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆811Updated this week
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆546Updated this week
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆518Updated this week
• pytorch / torcheval
  A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…
  ☆233Updated 4 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆374Updated last month
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆433Updated 3 months ago
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆514Updated 3 weeks ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆482Updated last year
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,201Updated this week
• google / grain
  Library for reading and processing ML training data.
  ☆447Updated this week
• google / fiddle
  ☆348Updated 3 weeks ago
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆586Updated last week
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆276Updated 11 months ago
• pytorch / torchcodec
  PyTorch video decoding
  ☆572Updated this week
• mit-ll-responsible-ai / hydra-zen
  Create powerful Hydra applications without the yaml files and boilerplate code.
  ☆385Updated this week
• apple / ml-sigma-reparam
  ☆303Updated 11 months ago
• Lightning-AI / utilities
  Common Python utilities and GitHub Actions in Lightning Ecosystem
  ☆56Updated this week
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆331Updated last year
• xl0 / lovely-tensors
  Tensors, for human consumption
  ☆1,252Updated last week
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆141Updated 3 months ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆183Updated 8 months ago
• huggingface / chug
  Minimal sharded dataset loaders, decoders, and utils for multi-modal document, image, and text datasets.
  ☆157Updated last year
• microsoft / Samba
  [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling
  ☆876Updated last month
• pytorch / ao
  PyTorch native quantization and sparsity for training and inference
  ☆2,072Updated this week
• KellerJordan / Muon
  Muon: An optimizer for hidden layers in neural networks
  ☆678Updated last week
• NVIDIA-Merlin / dataloader
  The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
  ☆418Updated last year