labmlai / labml
🔎 Monitor deep learning model training and hardware usage from your mobile phone 📱
☆2,125Updated last month
Alternatives and similar repositories for labml:
Users that are interested in labml are comparing it to the libraries listed below
- Official DeiT repository☆4,165Updated last year
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,718Updated 3 weeks ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,907Updated 9 months ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,278Updated last year
- PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO☆6,713Updated 8 months ago
- Example deep learning projects that use wandb's features.☆1,157Updated last month
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,228Updated this week
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,160Updated last week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆8,514Updated this week
- Code release for ConvNeXt model☆5,925Updated 2 years ago
- Foundation Architecture for (M)LLMs☆3,064Updated 11 months ago
- The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflo…☆804Updated 2 years ago
- PyTorch extensions for high performance and large scale training.☆3,283Updated 2 months ago
- View model summaries in PyTorch!☆2,730Updated last week
- Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)☆1,888Updated 7 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,812Updated last month
- Model interpretability and understanding for PyTorch☆5,145Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,590Updated last week
- SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners☆4,227Updated last year
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆22,208Updated 3 weeks ago
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,822Updated 3 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,095Updated last year
- StudioGAN is a Pytorch library providing implementations of representative Generative Adversarial Networks (GANs) for conditional/uncondi…☆3,452Updated 7 months ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆14,527Updated 8 months ago
- A PyTorch Library for Meta-learning Research☆2,738Updated 9 months ago
- An ultimately comprehensive paper list of Vision Transformer/Attention, including papers, codes, and related websites☆4,816Updated 7 months ago
- [NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang☆1,669Updated 2 years ago
- ☆11,108Updated 3 weeks ago
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,197Updated 3 years ago
- Scenic: A Jax Library for Computer Vision Research and Beyond☆3,478Updated 2 weeks ago