Alternatives and similar repositories for labml:

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

• clementchadebec / benchmark_VAE
  Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
  ☆1,863Updated 5 months ago
• lucidrains / x-transformers
  A concise but complete full-attention transformer with a set of promising experimental features from various papers
  ☆4,985Updated last week
• Lightning-Universe / lightning-bolts
  Toolbox of models, callbacks, and datasets for AI/ML researchers.
  ☆1,704Updated last week
• Lightning-AI / deep-learning-project-template
  Pytorch Lightning code guideline for conferences
  ☆1,245Updated last year
• libffcv / ffcv
  FFCV: Fast Forward Computer Vision (and other ML workloads!)
  ☆2,880Updated 7 months ago
• labmlai / annotated_deep_learning_paper_implementations
  🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, fee…
  ☆57,925Updated 4 months ago
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,655Updated last week
• kornia / kornia
  🐍 Geometric Computer Vision Library for Spatial AI
  ☆10,143Updated this week
• aimhubio / aim
  Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.
  ☆5,304Updated this week
• lucidrains / lion-pytorch
  🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch
  ☆2,080Updated last month
• Lightning-AI / torchmetrics
  Machine learning metrics for distributed, scalable PyTorch applications.
  ☆2,177Updated this week
• state-spaces / s4
  Structured state space sequence models
  ☆2,524Updated 6 months ago
• jettify / pytorch-optimizer
  torch-optimizer -- collection of optimizers for Pytorch
  ☆3,067Updated 9 months ago
• pytorch / captum
  Model interpretability and understanding for PyTorch
  ☆5,033Updated this week
• probml / pml2-book
  Probabilistic Machine Learning: Advanced Topics
  ☆1,423Updated last month
• dvgodoy / dl-visuals
  Over 200 figures and diagrams of the most popular deep learning architectures and layers FREE TO USE in your blog posts, slides, presenta…
  ☆1,407Updated 3 years ago
• Lightning-Universe / lightning-flash
  Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
  ☆1,746Updated last year
• google / learned_optimization
  ☆759Updated last week
• ashleve / lightning-hydra-template
  PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡
  ☆4,392Updated 5 months ago
• TylerYep / torchinfo
  View model summaries in PyTorch!
  ☆2,662Updated this week
• cmhungsteve / Awesome-Transformer-Attention
  An ultimately comprehensive paper list of Vision Transformer/Attention, including papers, codes, and related websites
  ☆4,716Updated 5 months ago
• CalculatedContent / WeightWatcher
  The WeightWatcher tool for predicting the accuracy of Deep Neural Networks
  ☆1,496Updated 4 months ago
• huggingface / accelerate
  🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…
  ☆8,178Updated this week
• microsoft / torchscale
  Foundation Architecture for (M)LLMs
  ☆3,038Updated 9 months ago
• IgorSusmelj / pytorch-styleguide
  An unofficial styleguide and best practices summary for PyTorch
  ☆1,948Updated 3 years ago
• dair-ai / ml-visuals
  🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.
  ☆14,073Updated last year
• rentruewang / koila
  Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
  ☆1,823Updated 3 weeks ago
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
  ☆28,807Updated last week
• facebookresearch / ConvNeXt
  Code release for ConvNeXt model
  ☆5,845Updated 2 years ago
• parrt / tensor-sensor
  The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflo…
  ☆801Updated 2 years ago