arogozhnikov / einopsLinks
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
ā9,304Updated 2 weeks ago
Alternatives and similar repositories for einops
Users that are interested in einops are comparing it to the libraries listed below
Sorting:
- š A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (iā¦ā9,348Updated this week
- Flax is a neural network library for JAX that is designed for flexibility.ā6,946Updated last week
- View model summaries in PyTorch!ā2,883Updated last week
- torch-optimizer -- collection of optimizers for Pytorchā3,154Updated last year
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.ā4,716Updated last week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorchā8,865Updated last week
- A concise but complete full-attention transformer with a set of promising experimental features from various papersā5,706Updated last month
- Machine learning metrics for distributed, scalable PyTorch applications.ā2,372Updated last week
- A small package to create visualizations of PyTorch execution graphsā3,469Updated 11 months ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)ā2,986Updated last year
- Model interpretability and understanding for PyTorchā5,481Updated last week
- PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ā”š„ā”ā5,033Updated last year
- Model summary in PyTorch similar to `model.summary()` in Kerasā4,065Updated last year
- PyTorch extensions for high performance and large scale training.ā3,387Updated 7 months ago
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.ā30,546Updated last week
- PyTorch code for Vision Transformers training with the Self-Supervised learning method DINOā7,343Updated last year
- Toolbox of models, callbacks, and datasets for AI/ML researchers.ā1,752Updated last week
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.ā6,267Updated 3 months ago
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.ā2,905Updated 5 months ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.ā3,291Updated last year
- š¦ Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorchā2,177Updated last year
- Hydra is a framework for elegantly configuring complex applicationsā10,005Updated last week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --ā¦ā35,941Updated this week
- A scikit-learn compatible neural network library that wraps PyTorchā6,139Updated 3 weeks ago
- JAX-based neural network libraryā3,136Updated this week
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.ā1,833Updated 3 weeks ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)ā7,977Updated last month
- An unofficial styleguide and best practices summary for PyTorchā2,001Updated 3 years ago
- Hopfield Networks is All You Needā1,880Updated 2 years ago
- ā12,087Updated 9 months ago