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)
9k stars
69 watching
350 forks
Language: Python
last commit: about 1 month ago
Linked from 3 awesome lists
chainercupydeep-learningeinopsjaxkerasnumpypytorchtensortensorflow
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 |