einops

arogozhnikov/einops

Deep learning operations reinvented (for pytorch, tensorflow, jax and others).

  • deep-learning
  • pytorch
  • tensorflow
  • numpy
  • cupy
  • tensor
  • jax
  • einops

More in Python

TensorFlowtensorflow/tensorflow

Open source software library for numerical computation using data flow graphs.

GeneralDeep learning
201.2k updated today

Transformershuggingface/transformers

A deep learning library containing thousands of pre-trained models on different tasks. The goto place for anything related to Large Language Models.

Deep learningLLMsNLPSpeech
168.1k updated today

PyTorchpytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

GeneralDeep learning
105k updated today

keraskeras-team/keras

High-level neural networks frontend for TensorFlow, CNTK and Theano.

GeneralDeep learning
64.4k updated 1d ago

Streamlitstreamlit/streamlit

Streamlit is an framework to create beautiful data apps in hours, not weeks.

GeneralDeep learningData & viz
46.1k updated today

RLlibray-project/ray

RLlib is an industry level, highly scalable RL library for tf and torch, based on Ray. It's used by companies like Amazon and Microsoft to solve real-world decision making problems at scale.

Deep learningRL
44.1k updated today

Colossal-AIhpcaitech/ColossalAI

An open-source deep learning system for large-scale model training and inference with high efficiency and low cost.

GeneralDeep learning
41.4k updated 2d ago

MindsDBmindsdb/mindsdb

Open Source framework to streamline use of neural networks.

GeneralDeep learning
39.8k updated 1d ago
All 443 in Python
Esc

Type to search every entry. ↑↓ to move, Enter to open.