Pinned Loading

1. NVIDIA/Megatron-LM NVIDIA/Megatron-LM Public

   Ongoing research training transformer models at scale

   Python 12.6k 2.9k

2. NVIDIA/TransformerEngine NVIDIA/TransformerEngine Public

   A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory…

   Python 2.5k 434

3. Image-Generation-models Image-Generation-models Public

   Generative models (GAN, VAE, Diffusion Models, Autoregressive Models) implemented with Pytorch, Pytorch_lightning and hydra.

   Python 59 11

4. PP-Schedule-Visualization PP-Schedule-Visualization Public

   Pipeline Parallelism Emulation and Visualization

   Python 43 3

5. PyTorch-Memory-Profiler PyTorch-Memory-Profiler Public

   Python 4