Wei Feng
@weifengpy
About
PyTorch Distributed, FSDP2, FlexShard
Skills & Technologies
Projects & Repositories
Recent public projects and repositories from this profile.
Megatron-FlexShard
PythonOngoing research training transformer models at scale
flex_shard
Flexible parameter sharding and distributed optimization utilities for PyTorch training
torchtitan
PythonA native PyTorch Library for large model training
autoparallel
An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.
test-infra
TypeScriptThis repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
verl
Pythonverl: Volcano Engine Reinforcement Learning for LLMs