Pinned Repositories
DeepBench
Benchmarking Deep Learning operations on different hardware
FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
flash-attention
Fast and memory-efficient exact attention
fusion-patterns-benchmark
llama
Inference code for LLaMA models
lumiere-pytorch
Implementation of Lumiere, SOTA text-to-video generation from Google Deepmind, in Pytorch
pipedream
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
torchdynamo_dashboard
TorchDynamo Dashboard Infra
tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
anijain2305's Repositories
anijain2305/tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
anijain2305/torchdynamo_dashboard
TorchDynamo Dashboard Infra
anijain2305/DeepBench
Benchmarking Deep Learning operations on different hardware
anijain2305/FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
anijain2305/flash-attention
Fast and memory-efficient exact attention
anijain2305/fusion-patterns-benchmark
anijain2305/llama
Inference code for LLaMA models
anijain2305/lumiere-pytorch
Implementation of Lumiere, SOTA text-to-video generation from Google Deepmind, in Pytorch
anijain2305/pipedream
anijain2305/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
anijain2305/tf_quantization
anijain2305/torchdynamo
anijain2305/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
anijain2305/torchrec
Pytorch domain library for recommendation systems
anijain2305/torchtune
A Native-PyTorch Library for LLM Fine-tuning
anijain2305/tvm_autoq
anijain2305/tvm_qnn_scripts
anijain2305/web-data
The repo to host all the web data including images for documents in dmlc projects.
anijain2305/xformers
Hackable and optimized Transformers building blocks, supporting a composable construction.
anijain2305/xla
Enabling PyTorch on Google TPU