Lists (1)
Stars
Tensors and Dynamic neural networks in Python with strong GPU acceleration
real time face swap and one-click video deepfake with only a single image
A high-throughput and memory-efficient inference and serving engine for LLMs
Local UI to run and train LLMs and diffusion models. Supports GGUF, MLX, Qwen3.8, DeepSeek-V4, MiniMax-H3, Gemma 4, FLUX and more.
Convert PDF to markdown + JSON quickly with high accuracy
DSPy: The framework for programming—not prompting—language models
You like pytorch? You like micrograd? You love tinygrad! ❤️
Official inference repo for FLUX.1 models
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Fast and memory-efficient exact attention
Graph Neural Network Library for PyTorch
Machine Learning Engineering Open Book
Janus-Series: Unified Multimodal Understanding and Generation Models
🚀 Level up your GitHub profile readme with customizable cards including LOC statistics!
TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. Tensor…
Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
Self-hosted bookmark manager that is designed be to be minimal, fast, and easy to set up using Docker.
Minimal and clean examples of machine learning algorithms implementations
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
Matplotlib styles for scientific plotting
AlphaFold 3 inference pipeline.
Efficiently computes derivatives of NumPy code.
A Python framework for GPU-accelerated simulation, robotics, and machine learning.
arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv