Pinned Repositories
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
export-playground
Simple experiments with torch.export
mtsc
A simple multi-thread server-client model using c10d
nccl
Optimized primitives for collective multi-GPU communication
parallel_blocks
PyTorch DTensor based basic parallel modules
pippy-device-map-playground
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
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.
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
kwen2501's Repositories
kwen2501/mtsc
A simple multi-thread server-client model using c10d
kwen2501/export-playground
Simple experiments with torch.export
kwen2501/nccl
Optimized primitives for collective multi-GPU communication
kwen2501/parallel_blocks
PyTorch DTensor based basic parallel modules
kwen2501/pippy-device-map-playground
kwen2501/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
kwen2501/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.
kwen2501/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.