Ke Wen's repositories
export-playground
Simple experiments with torch.export
nccl
Optimized primitives for collective multi-GPU communication
Language:C++NOASSERTION000
Language:Python000
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:C++NOASSERTION000
test-infra
This repository hosts code that supports the testing infrastructure for the main PyTorch repo. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
Language:TypeScriptNOASSERTION000
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Language:PythonApache-2.0000