kshitij12345 Goto Github PK

algorithm-practice

array-api

RFC document, tooling and other content related to the array API standard

arrow

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

audio

Data manipulation and transformation for audio signal processing, powered by PyTorch

benchmark

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

boneos

BoneOS Kernel and Operating System Source Tree

chainer

A flexible framework of neural networks for deep learning

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.

clusterer

A package for easier interface and setup for pipelining in distributed cluster.

computer-vision-algorithms

Computer Vision Algorithms

contextualspellcheck

✔️Contextual word checker for better suggestions

cub

Cooperative primitives for CUDA C++.

cupy

NumPy & SciPy for GPU

dask-ml

Scalable Machine Learn with Dask

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.

graph-algorithms

Algorithms on Graphs

graph-sched

halide

a language for fast, portable data-parallel computation

haskell-playground

Learning and Playing around in Haskell - Beginner Stuff

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

incubator-tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

jumpy

A Python interface for ND4J: A Numpy Array Wrapper for the JVM

keops

KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows

kornia

Open Source Differentiable Computer Vision Library for PyTorch

kshitij12345.github.io

Personal Site

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.