kshitij12345
Open Source and Machine Learning Enthusiast | Contributed to PyTorch, Apache MXNet, Chainer, Apache Arrow
NvidiaThe Netherlands
Pinned Repositories
audio
Data manipulation and transformation for audio signal processing, powered by PyTorch
clusterer
A package for easier interface and setup for pipelining in distributed cluster.
contextualSpellCheck
✔️Contextual word checker for better suggestions
graph-sched
Halide
a language for fast, portable data-parallel computation
keops
KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
kshitij12345.github.io
Personal Site
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
reverseautodiff
torchnnprofiler
Context Manager to profile the forward and backward times of PyTorch's nn.Module
kshitij12345's Repositories
kshitij12345/torchnnprofiler
Context Manager to profile the forward and backward times of PyTorch's nn.Module
kshitij12345/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
kshitij12345/kshitij12345.github.io
Personal Site
kshitij12345/python-plugin
kshitij12345/ao
PyTorch native quantization and sparsity for training and inference
kshitij12345/arrow
Apache Arrow is a multi-language toolbox for accelerated data interchange and in-memory processing
kshitij12345/benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
kshitij12345/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.
kshitij12345/cub
Cooperative primitives for CUDA C++.
kshitij12345/cupy
NumPy & SciPy for GPU
kshitij12345/examples
📝 Examples of experiment tracking, model registry, data versioning, and monitoring machine learning model training live in Neptune.ai
kshitij12345/functorch
functorch is a prototype of JAX-like composable function transforms for PyTorch.
kshitij12345/ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
kshitij12345/lightning
Deep learning framework to train, deploy, and ship AI products Lightning fast.
kshitij12345/lightning-thunder
Source to source compiler for PyTorch. It makes PyTorch programs faster on single accelerators and distributed.
kshitij12345/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.
kshitij12345/MatX
An efficient C++17 GPU numerical computing library with Python-like syntax
kshitij12345/method-chainer
Lazily evaluate method chain in Python
kshitij12345/notebooks
kshitij12345/numpy
The fundamental package for scientific computing with Python.
kshitij12345/nussl
A flexible source separation library in Python
kshitij12345/opacus
Training PyTorch models with differential privacy
kshitij12345/pytorch-transformers
👾 A library of state-of-the-art pretrained models for Natural Language Processing (NLP)
kshitij12345/Quansight-website
💻 Source code for Quansight Consulting and Labs websites
kshitij12345/skorch
A scikit-learn compatible neural network library that wraps PyTorch
kshitij12345/thrust
The C++ parallel algorithms library.
kshitij12345/torch-build
Collection of scripts to build PyTorch and the domain libraries from source.
kshitij12345/wg21_p2300_std_execution
`std::execution`, the proposed C++ framework for asynchronous and parallel programming.
kshitij12345/xla
Enabling PyTorch on Google TPU
kshitij12345/zig
General-purpose programming language and toolchain for maintaining robust, optimal, and reusable software.