Chaitanya Sri Krishna Lolla's repositories
cupy
A NumPy-compatible array library accelerated by CUDA
Megatron-LM
Ongoing research training transformer language models at scale, including: BERT & GPT-2
nn-tools
Some useful neural network tools and some deep learning experiments for quick prototyping and testing.
vllm
A high-throughput and memory-efficient inference and serving engine for LLMs
my-pytorch-experiments
Contains some of my pytorch experiments and practise
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.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
vision
Datasets, Transforms and Models specific to Computer Vision
training_results_v0.7
Training v0.7 results
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
ieee-sps-ml-tutorial
IEEE SPS Machine Learning Training tutorial
nccl-rccl-parser
Tool to run rccl-tests/nccl-tests based on from an application and gather performance.
transformers
🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
ClassyVision
An end-to-end PyTorch framework for image and video classification
VMZ
VMZ: Model Zoo for Video Modeling
DeepLearningExamples
Deep Learning Examples
dlrm
An implementation of a deep learning recommendation model (DLRM)
DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
detectron2
Detectron2 is FAIR's next-generation research platform for object detection and segmentation.
maskrcnn-benchmark
Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
OpenVX-sample-impl
OpenVX sample implementation
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
unix-cheatsheet
Notes on Shell scripting and Unix Operating system.
py-mi
Pytorch Profling tools and analysis
fruit-classification-pytorch
Fruit classification Kaggle dataset using pytorch