Stars
Framework for Algorithmic Correctness Testing of Operators
A powerful AI coding agent. Built for the terminal.
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
PyTorch native quantization and sparsity for training and inference
depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.
A PyTorch native platform for training generative AI models
Building a quick conversation-based search demo with Lepton AI.
TensorDict is a pytorch dedicated tensor container.
The definitive Web UI for local AI, with powerful features and easy setup.
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
An open-source framework for training large multimodal models.
The simplest, fastest repository for training/finetuning medium-sized GPTs.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
ASDL: Automatic Second-order Differentiation Library for PyTorch
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single C++ process.
ROS package that can filter geometry defined in URDF models from Kinect depth images. Can also preprocess data for the OpenNI tracker, to remove backgrounds, robots etc.
[CVPR'22] NICE-SLAM: Neural Implicit Scalable Encoding for SLAM
Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.
C++ wrappers for SIMD intrinsics and parallelized, optimized mathematical functions (SSE, AVX, AVX512, NEON, SVE))
Visual Inertial Odometry with SLAM capabilities and 3D Mesh generation.