Repository Showcase
A curated collection of interesting GitHub repositories
NVIDIA/Megatron-LM
GPU-optimized library for training massive transformer models at scale
Related repos
- Python framework for fast GPU simulation and graphics
- Nanotron: minimalistic library for pretraining transformer models with simple API and scalable parallelism
- Extension of Transformers for optimizing and running models on specific hardware
- NVIDIA Isaac GR00T lets robots use language and images to do tasks in different environments
- transformers library for working with machine learning models
- Open-source robotic arm kit with NVIDIA Jetson integration
- interactive NVIDIA GPU process viewer
- Run 70B LLMs on a 4GB GPU with layer-wise inference and memory optimization, quantization optional
- scans LLMs for vulnerabilities like prompt injection, hallucination, and data leaks
- serving and scaling generative AI models across distributed GPUs
