Iman Tabrizian's starred repositories
flash-attention
Fast and memory-efficient exact attention
nn-zero-to-hero
Neural Networks: Zero to Hero
the-incredible-pytorch
The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.
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.
libcudacxx
[ARCHIVED] The C++ Standard Library for your entire system. See https://github.com/NVIDIA/cccl
Essentials-of-Compilation
A book about compiling Racket and Python to x86-64 assembly
onnxruntime-genai
Generative AI extensions for onnxruntime
extending-jax
Extending JAX with custom C++ and CUDA code
multi-core-python
Enabling CPython multi-core parallelism via subinterpreters.
cuda-checkpoint
CUDA checkpoint and restore utility
extrainterpreters
Utilities for using Python's PEP 554 subinterpreters
vscode-micromamba
A VSCode extension to generate development environments using micromamba and conda-forge package repository