kshitij12345

torchnnprofiler

Context Manager to profile the forward and backward times of PyTorch's nn.Module

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

kshitij12345.github.io

Personal Site

python-plugin

arrow

Apache Arrow is a multi-language toolbox for accelerated data interchange and in-memory processing

benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

ChainRulesCore.jl

AD-backend agnostic system defining custom forward and reverse mode rules. This is the light weight core to allow you to define rules for your functions in your packages, without depending on any particular AD system.

cub

Cooperative primitives for CUDA C++.

cupy

NumPy & SciPy for GPU

examples

📝 Examples of experiment tracking, model registry, data versioning, and monitoring machine learning model training live in Neptune.ai

functorch

functorch is a prototype of JAX-like composable function transforms for PyTorch.

ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

incubator-mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more

lightning

Deep learning framework to train, deploy, and ship AI products Lightning fast.

lightning-thunder

Source to source compiler for PyTorch. It makes PyTorch programs faster on single accelerators and distributed.

lit-gpt

Hackable implementation of state-of-the-art open-source LLMs based on nanoGPT. Supports flash attention, 4-bit and 8-bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.

MatX

An efficient C++17 GPU numerical computing library with Python-like syntax

method-chainer

Lazily evaluate method chain in Python

notebooks

numpy

The fundamental package for scientific computing with Python.

nussl

A flexible source separation library in Python

opacus

Training PyTorch models with differential privacy

pytorch-transformers

👾 A library of state-of-the-art pretrained models for Natural Language Processing (NLP)

Quansight-website

💻 Source code for Quansight Consulting and Labs websites

skorch

A scikit-learn compatible neural network library that wraps PyTorch

thrust

The C++ parallel algorithms library.

torch-build

Collection of scripts to build PyTorch and the domain libraries from source.

wg21_p2300_std_execution

`std::execution`, the proposed C++ framework for asynchronous and parallel programming.

xla

Enabling PyTorch on Google TPU

zig

General-purpose programming language and toolchain for maintaining robust, optimal, and reusable software.