Tensor library for machine learning
Solve puzzles. Improve your pytorch
Cosmos-RL is a flexible and scalable Reinforcement Learning framework
A lightweight vLLM implementation built from scratch
Our first fully AI generated deep learning system
Mooncake is the serving platform for Kimi
Ongoing research training transformer models at scale
Python library for defining and optimizing mathematical expressions
Practice implementing softmax, attention, GPT-2 and more
A course of learning LLM inference serving on Apple Silicon
CV, NLP, LLM project applications, and advanced engineering deployment
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Deep learning optimization library: makes distributed training easy
Large-language-model & vision-language-model based on Linear Attention
Deep and online learning with spiking neural networks in Python
Mars is a tensor-based unified framework for large-scale data
Transformer related optimization, including BERT, GPT
Code for Tensorflow Machine Learning Cookbook
Large-scale xAI model for local inference with SGLang, Grok-2.5
685B model with improved agents and consistency