Popular repositories Loading

1. fx_graph_converter fx_graph_converter Public

   Quick tool to extract FX Graph Modules from PyTorch networks and convert them into NetworkX graphs.

   1

2. stablediffusion stablediffusion Public

   Forked from alpha0422/stablediffusion

   High-Resolution Image Synthesis with Latent Diffusion Models

   Python

3. NeMo_llama NeMo_llama Public

   Forked from NVIDIA-NeMo/NeMo

   NeMo: a toolkit for conversational AI

   Python

4. TransformerEngine TransformerEngine Public

   Forked from NVIDIA/TransformerEngine

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

   Python

5. Megatron-LM Megatron-LM Public

   Forked from NVIDIA/Megatron-LM

   Ongoing research training transformer models at scale

   Python