Wei Feng
GitHub

Wei Feng

@weifengpy

View GitHub Profile

About

PyTorch Distributed, FSDP2, FlexShard

Location US
Company PyTorch
Followers 101
Following 151
Public Repositories 27
Expertise

Skills & Technologies

Python TypeScript
GitHub Work

Projects & Repositories

Recent public projects and repositories from this profile.

Megatron-FlexShard

Python

Ongoing research training transformer models at scale

⭐ 0 Forks: 0

flex_shard

Flexible parameter sharding and distributed optimization utilities for PyTorch training

⭐ 0 Forks: 0
Repository View Project

torchtitan

Python

A native PyTorch Library for large model training

⭐ 0 Forks: 0

autoparallel

An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.

⭐ 1 Forks: 0
Repository View Project

test-infra

TypeScript

This 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.

⭐ 0 Forks: 0
TypeScript View Project

verl

Python

verl: Volcano Engine Reinforcement Learning for LLMs

⭐ 0 Forks: 0