lcskrishna

fruit-classification-pytorch

Fruit classification Kaggle dataset using pytorch

nccl-rccl-parser

Tool to run rccl-tests/nccl-tests based on from an application and gather performance.

py-mi

Pytorch Profling tools and analysis

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

ClassyVision

An end-to-end PyTorch framework for image and video classification

cupy

A NumPy-compatible array library accelerated by CUDA

DeepLearningExamples

Deep Learning Examples

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.

dlrm

An implementation of a deep learning recommendation model (DLRM)

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

hipify_torch

ieee-sps-ml-tutorial

IEEE SPS Machine Learning Training tutorial

lcskrishna.github.io

maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

Megatron-LM

Ongoing research training transformer language models at scale, including: BERT & GPT-2

my-cuda-experiments

my-pytorch-experiments

Contains some of my pytorch experiments and practise

nn-tools

Some useful neural network tools and some deep learning experiments for quick prototyping and testing.

OpenVX-sample-impl

OpenVX sample implementation

pytorch-autograd-profile-analyzer

reinforcement-dl

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.

training_results_v0.7

Training v0.7 results

transformers

🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.

unix-cheatsheet

Notes on Shell scripting and Unix Operating system.

vision

Datasets, Transforms and Models specific to Computer Vision

vllm

A high-throughput and memory-efficient inference and serving engine for LLMs

VMZ

VMZ: Model Zoo for Video Modeling