pytorch / captum
Model interpretability and understanding for PyTorch
☆5,173Updated this week
Alternatives and similar repositories for captum:
Users that are interested in captum are comparing it to the libraries listed below
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,845Updated 2 months ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,238Updated this week
- torch-optimizer -- collection of optimizers for Pytorch☆3,099Updated last year
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,118Updated this week
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,648Updated last week
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,279Updated last year
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,719Updated this week
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,293Updated this week
- Tips for releasing research code in Machine Learning (with official NeurIPS 2020 recommendations)☆2,700Updated last year
- A small package to create visualizations of PyTorch execution graphs☆3,335Updated 3 months ago
- PyTorch extensions for high performance and large scale training.☆3,293Updated this week
- View model summaries in PyTorch!☆2,740Updated this week
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,038Updated last year
- Pytorch Lightning code guideline for conferences☆1,258Updated last year
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,917Updated 9 months ago
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆8,608Updated this week
- A scikit-learn compatible neural network library that wraps PyTorch☆6,000Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,607Updated this week
- A hyperparameter optimization framework☆11,719Updated this week
- Official DeiT repository☆4,173Updated last year
- A curated list of awesome self-supervised methods☆6,257Updated 9 months ago
- SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners☆4,239Updated last year
- An unofficial styleguide and best practices summary for PyTorch☆1,971Updated 3 years ago
- A learning rate range test implementation in PyTorch☆952Updated 4 months ago
- [CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize …☆1,870Updated last year
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,929Updated 2 months ago
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.☆29,263Updated this week
- PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722☆4,942Updated 3 months ago
- PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)☆3,244Updated last year
- Code for visualizing the loss landscape of neural nets☆2,953Updated 3 years ago