Alternatives and similar repositories for stoke

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

• pytorch / torchdistx
  Torch Distributed Experimental
  ☆116Updated 10 months ago
• iKernels / transformers-lightning
  A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…
  ☆47Updated 2 years ago
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 3 years ago
• tchaton / lightning-hydra-seed
  Pytorch Lightning seed project with hydra
  ☆18Updated 4 years ago
• sdpython / mlprodict
  Productionize machine learning predictions, with ONNX or without
  ☆65Updated last year
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆44Updated 3 years ago
• lf1-io / padl
  Functional deep learning
  ☆108Updated 2 years ago
• Lightning-AI / ecosystem-ci
  Automate issue discovery for your projects against Lightning nightly and releases.
  ☆46Updated last month
• Yard1 / ray-skorch
  Distributed skorch on Ray Train
  ☆57Updated 2 years ago
• gridai / gridai
  Placeholder for the opensource Grid AI components
  ☆44Updated 3 years ago
• pytorch-tpu / examples
  This repository contains example code to build models on TPUs
  ☆30Updated 2 years ago
• TezRomacH / layer-to-layer-pytorch
  PyTorch implementation of L2L execution algorithm
  ☆107Updated 2 years ago
• pietrolesci / energizer
  An active learning library for Pytorch based on Lightning-Fabric.
  ☆79Updated last year
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year
• Borda / pyDeprecate
  Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.
  ☆51Updated 2 months ago
• ray-project / distml
  Distributed ML Optimizer
  ☆32Updated 3 years ago
• tmbdev-archive / pytorch-imagenet-wds
  ☆25Updated 4 years ago
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆211Updated last year
• theblackcat102 / H5Record
  Large dataset storage format for Pytorch
  ☆45Updated 3 years ago
• HomebrewML / HomebrewNLP-torch
  A case study of efficient training of large language models using commodity hardware.
  ☆69Updated 2 years ago
• JulesBelveze / bert-squeeze
  🛠️ Tools for Transformers compression using PyTorch Lightning ⚡
  ☆83Updated 7 months ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆157Updated this week
• nestordemeure / flaxOptimizers
  A collection of optimizers, some arcane others well known, for Flax.
  ☆29Updated 3 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• saturncloud / dask-pytorch-ddp
  dask-pytorch-ddp is a Python package that makes it easy to train PyTorch models on dask clusters using distributed data parallel.
  ☆59Updated 4 years ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆37Updated 2 years ago
• fastai / docments
  Document parameters using comments
  ☆10Updated 3 years ago
• Lightning-Universe / Training-Studio_app
  Lightning HPO & Training Studio App
  ☆18Updated 2 years ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated 2 years ago