Pinned Repositories
aml-nlp
homework of coursera nlp course. https://www.coursera.org/learn/language-processing/home/welcome
clrsPython
gameofdimension.github.io
inception-java-client
java client for tensorflow serving's inception example
java-bert-predict
turn bert pretrain checkpoint into saved model for a feature extracting demo in java
jni-faiss
java native interface for faiss
policy-gradient-pong
tensorflow implementation of Andrej Karpathy's blog about reinforcement learning. http://karpathy.github.io/2016/05/31/rl/
vllm
A high-throughput and memory-efficient inference and serving engine for LLMs
vllm-cn
演示 vllm 对中文大语言模型的神奇效果
xgboost_explainer
A tool for analyzing feature importance of xgboost model. idea came from R version xgboostExplainer
gameofdimension's Repositories
gameofdimension/limulidae
benchmark gpu/npu flops and bandwidth
gameofdimension/kapok
distributed inference for DiTs, in plain pytorch
gameofdimension/rope-cut
gameofdimension/tiny-grpo
A torch native grpo training example
gameofdimension/6.S184-lab
gameofdimension/ascend-picotron
Minimalistic 4D-parallelism distributed training framework for education purpose. Adapting to Ascend NPU.
gameofdimension/cache-dit
🤗A Training-free and Easy-to-use Cache Acceleration Toolbox for DiTs: DBCache, DBPrune, TaylorSeer, FBCache, etc🔥
gameofdimension/cs336-assignment1
2025 spring cs336 assignment1
gameofdimension/cs336-assignment2
2025 spring cs336 assignment2
gameofdimension/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.
gameofdimension/dpm-flow-match
A demo on how to apply DPM to Flow Match models
gameofdimension/flux
A fast communication-overlapping library for tensor/expert parallelism on GPUs.
gameofdimension/FluxKits
gameofdimension/gpu_benchmark
NPU/GPU peak FLOPs benchmark
gameofdimension/hfans
gameofdimension/lectures
Material for gpu-mode lectures
gameofdimension/mamba-minimal
gameofdimension/my-diffusion
as a way of studying diffusion model
gameofdimension/my-transformer
explore transformer
gameofdimension/ParaAttention
https://wavespeed.ai/ Context parallel attention that accelerates DiT model inference with dynamic caching
gameofdimension/prodigy
The Prodigy optimizer and its variants for training neural networks.
gameofdimension/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
gameofdimension/ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
gameofdimension/rl-z2h
gameofdimension/Sparse-VideoGen
[ICML2025] Sparse VideoGen: Accelerating Video Diffusion Transformers with Spatial-Temporal Sparsity
gameofdimension/torchtitan
A native PyTorch Library for large model training
gameofdimension/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
gameofdimension/tutorials
PyTorch tutorials.
gameofdimension/verl
verl: Volcano Engine Reinforcement Learning for LLMs
gameofdimension/vllm-ascend
Community maintained hardware plugin for vLLM on Ascend