einops

Tensor library

A flexible and powerful library for performing tensor operations in various deep learning frameworks.

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

GitHub

9k stars
69 watching
350 forks
Language: Python
last commit: about 1 month ago
Linked from 3 awesome lists

chainercupydeep-learningeinopsjaxkerasnumpypytorchtensortensorflow

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
pytorch/pytorch A Python library providing tensors and dynamic neural networks with strong GPU acceleration 83,959
pkmital/tensorflow_tutorials A collection of examples and tutorials illustrating various applications and use cases of TensorFlow in Python 5,642
lanpa/tensorboardx A tool for visualizing and logging the progress of machine learning models during training 7,870
skelsec/pypykatz An implementation of Mimikatz in pure Python for parsing Windows secrets and registry data. 2,879
wkentaro/pytorch-for-numpy-users Provides a PyTorch interface for working with NumPy arrays and tensors. 687
huggingface/transformers A collection of pre-trained machine learning models for various natural language and computer vision tasks, enabling developers to fine-tune and deploy these models on their own projects. 135,022
nvidia/minkowskiengine An auto-differentiation library for sparse tensors used in computer vision and deep learning applications. 2,485
qubvel-org/segmentation_models.pytorch A PyTorch library for building and training neural networks for image segmentation tasks. 9,696
codertimo/bert-pytorch An implementation of Google's 2018 BERT model in PyTorch, allowing pre-training and fine-tuning for natural language processing tasks 6,222
tensorpack/tensorpack A high-performance neural network training interface for TensorFlow that focuses on speed and flexibility. 6,303
bytedance/byteps A high-performance distributed deep learning framework supporting multiple frameworks and networks 3,630
tensorflow/lingvo A software framework for building sequence models using neural networks in TensorFlow 2,816
huggingface/safetensors A format for storing tensors in a safe and efficient manner. 2,890
thtrieu/darkflow Tools and scripts for training and deploying real-time object detection models using TensorFlow 6,132
vahidk/effectivetensorflow A comprehensive tutorial and best practices guide for the TensorFlow deep learning framework 8,624