Pinned Repositories
kineto
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
PyProf
A GPU performance profiling tool for PyTorch models
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
pytorch-lightning
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
tutorials
PyTorch tutorials.
pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
TensorRT-LLM
TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT-LLM also contains components to create Python and C++ runtimes that execute those TensorRT engines.
opencompass
OpenCompass is an LLM evaluation platform, supporting a wide range of models (Llama3, Mistral, InternLM2,GPT-4,LLaMa2, Qwen,GLM, Claude, etc) over 100+ datasets.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
tutorials
PyTorch tutorials.
gaoteng-git's Repositories
gaoteng-git/kineto
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
gaoteng-git/PyProf
A GPU performance profiling tool for PyTorch models
gaoteng-git/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
gaoteng-git/pytorch-lightning
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
gaoteng-git/tutorials
PyTorch tutorials.