Pinned Loading

1. pytorch/pytorch pytorch/pytorch Public

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python 105k 32.5k

2. flex_shard flex_shard Public

   Forked from meta-pytorch/flex_shard

   Flexible parameter sharding and distributed optimization utilities for PyTorch training

   Python

3. torchtitan-flexshard torchtitan-flexshard Public

   Forked from pytorch/torchtitan

   A native PyTorch Library for large model training

   Python

4. Megatron-flexshard Megatron-flexshard Public

   Forked from NVIDIA/Megatron-LM

   Ongoing research training transformer models at scale

   Python