Alternatives and similar repositories for LitModels

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

• Lightning-AI / utilities
  Common Python utilities and GitHub Actions in Lightning Ecosystem
  ☆56Updated this week
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 11 months ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆141Updated 4 months ago
• google-deepmind / tf2jax
  ☆114Updated this week
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 10 months ago
• Lightning-Universe / Training-Studio_app
  Lightning HPO & Training Studio App
  ☆18Updated 2 years ago
• 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
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆121Updated 10 months ago
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆70Updated 2 weeks ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆110Updated 3 weeks ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆269Updated last week
• SkafteNicki / pl_crossvalidate
  ☆61Updated last year
• speediedan / finetuning-scheduler
  A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.
  ☆63Updated 2 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆102Updated 5 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months ago
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆111Updated 5 months ago
• lucidrains / agent-attention-pytorch
  Implementation of Agent Attention in Pytorch
  ☆90Updated 10 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆199Updated 4 months ago
• lucidrains / gotennet-pytorch
  Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch
  ☆62Updated 2 months ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 2 years ago
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆41Updated 3 months ago
• fkodom / python-repo-template
  Template repo for Python projects, especially those focusing on machine learning and/or deep learning.
  ☆15Updated 2 weeks ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• jiaweizzhao / ZerO-initialization
  ☆74Updated 2 years ago
• Charl-AI / stochastic-caching
  A tiny library for stochastic dataset caching in PyTorch.
  ☆42Updated last year
• jxtngx / lightning-lab
  deep learning with pytorch lightning
  ☆1Updated 7 months ago
• lucidrains / Adan-pytorch
  Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
  ☆252Updated 2 years ago
• twsl / whos-there
  The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends
  ☆77Updated 6 months ago
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆99Updated 9 months ago
• lucidrains / sinkhorn-router-pytorch
  Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise
  ☆35Updated 9 months ago