BladeDISC

tensorflow

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

human-eval

Code for the paper "Evaluating Large Language Models Trained on Code"

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

torch-mlir

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.

torchbenchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

vllm

A high-throughput and memory-efficient inference and serving engine for LLMs