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
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 9 months 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…
  ☆229Updated 4 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆100Updated 5 months ago
• twsl / whos-there
  The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends
  ☆77Updated 5 months ago
• jxtngx / lightning-lab
  deep learning with pytorch lightning
  ☆1Updated 6 months ago
• Lightning-Universe / Training-Studio_app
  Lightning HPO & Training Studio App
  ☆18Updated 2 years ago
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆68Updated 3 weeks ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 2 years 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 9 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 10 months ago
• google-deepmind / tf2jax
  ☆114Updated this week
• jiaweizzhao / ZerO-initialization
  ☆74Updated 2 years ago
• pietrolesci / energizer
  An active learning library for Pytorch based on Lightning-Fabric.
  ☆79Updated last year
• LumenPallidium / jepa
  Experiments in Joint Embedding Predictive Architectures (JEPAs).
  ☆40Updated last year
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆111Updated 4 months ago
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆40Updated 2 months ago
• sayakpaul / keras-xla-benchmarks
  Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
  ☆37Updated last year
• lucidrains / medical-ai-experiments
  A repository to house some personal attempts to beat some state-of-the-art for medical datasets
  ☆98Updated last year
• lucidrains / flash-genomics-model
  My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…
  ☆53Updated last year
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆91Updated last month
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆90Updated 3 years ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆141Updated 3 months ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆60Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆100Updated 4 months ago
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆259Updated this week
• pgniewko / forward_forward_vhts
  The Forward-Forward Algorithm for Drug Discovery
  ☆34Updated 2 years ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆88Updated 10 months ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated last year