interactive-keras-captioning

Tensors and Dynamic neural networks in Python with strong GPU acceleration

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

The fastai deep learning library

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RNN and transformer - great performance, linear time, constant space (no kv-cache), fast training, infinite ctx_len, and free sentence embedding.

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Open3D: A Modern Library for 3D Data Processing

NumPy & SciPy for GPU

An open-source, low-code machine learning library in Python

A PyTorch implementation of the Transformer model in "Attention is All You Need".