Alternatives and similar repositories for sweeps:

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

• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆210Updated last year
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆194Updated 2 months ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆115Updated 8 months ago
• google-deepmind / tf2jax
  ☆114Updated last week
• srush / torch-queue
  ☆67Updated last year
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆71Updated 3 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆155Updated 2 years ago
• twsl / whos-there
  The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends
  ☆77Updated 4 months ago
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆67Updated last year
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆109Updated 2 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆89Updated 3 years ago
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆172Updated 9 months ago
• google-deepmind / jaxline
  ☆155Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆134Updated 5 months ago
• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆78Updated 4 years ago
• rwightman / efficientnet-jax
  EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
  ☆128Updated last year
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆263Updated 3 years ago
• google-deepmind / tensor_annotations
  Annotating tensor shapes using Python types
  ☆159Updated last year
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆337Updated 2 weeks ago
• NERSC / slurm-ray-cluster
  ☆55Updated last year
• google / praxis
  ☆186Updated 2 weeks ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆206Updated last year
• pengzhenghao / use-ray-with-slurm
  A brief tutorial on how to use ray/rllib/tun in slurm cluster.
  ☆33Updated 4 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 9 months ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆140Updated 2 months ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated 2 years ago
• facebookresearch / moolib
  A library for distributed ML training with PyTorch
  ☆366Updated 2 years ago
• awaelchli / pytorch-lightning-snippets
  A collection of code snippets for my PyTorch Lightning projects
  ☆107Updated 4 years ago