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 last week
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• sayakpaul / keras-xla-benchmarks
  Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
  ☆37Updated last year
• lucidrains / transformer-directed-evolution
  Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster
  ☆70Updated 2 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 11 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…
  ☆235Updated 6 months ago
• Charl-AI / stochastic-caching
  A tiny library for stochastic dataset caching in PyTorch.
  ☆43Updated last year
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆143Updated 5 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• indri-voice / vit.triton
  VIT inference in triton because, why not?
  ☆30Updated last year
• johnma2006 / candle
  Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.
  ☆50Updated last year
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆124Updated 11 months ago
• google-deepmind / tf2jax
  ☆114Updated last week
• lucidrains / medical-ai-experiments
  A repository to house some personal attempts to beat some state-of-the-art for medical datasets
  ☆99Updated last year
• pytorch-labs / monarch
  PyTorch Single Controller
  ☆325Updated this week
• PrincetonUniversity / multi_gpu_training
  ☆345Updated 4 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆42Updated 7 months ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆183Updated 10 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆279Updated last year
• m1balcerak / EnergyMatching
  This is the official repository for the paper: "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling".
  ☆47Updated 2 weeks ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆106Updated 7 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆521Updated this week
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆42Updated 4 months ago
• Lightning-Universe / Training-Studio_app
  Lightning HPO & Training Studio App
  ☆18Updated 2 years ago
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆62Updated last week
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆19Updated last year
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆385Updated 3 months ago
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆111Updated 6 months ago
• lucidrains / genetic-algorithm-pytorch
  Toy genetic algorithm in Pytorch
  ☆53Updated 2 months ago