Stars
FlashMLA: Efficient MLA decoding kernels
The open and composable observability and data visualization platform. Visualize metrics, logs, and traces from multiple sources like Prometheus, Loki, Elasticsearch, InfluxDB, Postgres and many mo…
PyTorch native quantization and sparsity for training and inference
A minimal GPU design in Verilog to learn how GPUs work from the ground up
A Data Streaming Library for Efficient Neural Network Training
Mobile ALOHA: Learning Bimanual Mobile Manipulation with Low-Cost Whole-Body Teleoperation
A high-throughput and memory-efficient inference and serving engine for LLMs
Qdrant - High-performance, massive-scale Vector Database and Vector Search Engine for the next generation of AI. Also available in the cloud https://cloud.qdrant.io/
Pytorch implementation of DoReMi, a method for optimizing the data mixture weights in language modeling datasets
Development repository for the Triton language and compiler
Fast and memory-efficient exact attention
Fast, differentiable sorting and ranking in PyTorch
Python library for converting Python calculations into rendered latex.
Code accompanying the paper "Learning Agile Robotic Locomotion Skills by Imitating Animals"
Protocol Buffers - Google's data interchange format
An implementation of a deep learning recommendation model (DLRM)
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.