pytorch

Tensores e redes neurais dinâmicas em Python com forte aceleração por GPU

Um conjunto de exemplos usando PyTorch em Visão, Texto, Aprendizado por Reforço, etc.

Datasets, transformações e modelos específicos para Visão Computacional

Tutoriais de PyTorch.

Uma plataforma nativa PyTorch para treinar modelos de IA generativa

IA no dispositivo para mobile, embarcados e edge para PyTorch

Biblioteca de alto nível para auxiliar no treinamento e avaliação de redes neurais em PyTorch de forma flexível e transparente.

Uma biblioteca PyTorch modular, focada em primitivas e em Python para Aprendizado por Reforço.

Compilador PyTorch/TorchScript/FX para GPUs NVIDIA usando TensorRT

Quantização nativa em PyTorch para treinamento e inferência.

Manipulação e transformação de dados para processamento de sinais de áudio, com tecnologia PyTorch

Habilitando PyTorch em dispositivos XLA (por exemplo, Google TPU)

FB (Facebook) + GEMM (Multiplicação Geral de Matriz-Matriz) - https://code.fb.com/ml-applications/fbgemm/

Biblioteca de comunicações coletivas com várias primitivas para treinamento em múltiplas máquinas.

Submissão para https://pytorch.org/hub/

Extensões C++ no PyTorch

Biblioteca de informações de CPU (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS).

TorchBench é uma coleção de benchmarks de código aberto usados para avaliar o desempenho do PyTorch.

TensorDict é um contêiner de tensores dedicado ao pytorch.

Uma biblioteca de perfilamento de CPU+GPU que fornece acesso a rastreamentos de linha do tempo e contadores de desempenho de hardware.

Um DSL embutido em Python que facilita a escrita de kernels de ML rápidos e escaláveis com o mínimo de código boilerplate.