Labman42 / JetEngineLinks
A lightweight Inference Engine built for block diffusion models
☆40Updated last month
Alternatives and similar repositories for JetEngine
Users that are interested in JetEngine are comparing it to the libraries listed below
Sorting:
- ☆137Updated last week
- ☆118Updated 4 months ago
- ☆63Updated 6 months ago
- Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…☆197Updated 2 months ago
- TraceRL & TraDo-8B: Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models☆398Updated last month
- PoC for "SpecReason: Fast and Accurate Inference-Time Compute via Speculative Reasoning" [NeurIPS '25]☆61Updated 3 months ago
- Spectral Sphere Optimizer☆90Updated 2 weeks ago
- ☆220Updated 2 months ago
- paper list, tutorial, and nano code snippet for Diffusion Large Language Models.☆152Updated last week
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆288Updated 2 months ago
- ☆129Updated 7 months ago
- qwen-nsa☆87Updated 3 months ago
- A collection of papers on discrete diffusion models☆168Updated 7 months ago
- [ASPLOS'26] Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Drafter☆130Updated last month
- Official PyTorch implementation of the paper "Accelerating Diffusion Large Language Models with SlowFast Sampling: The Three Golden Princ…☆39Updated 6 months ago
- ☆55Updated 7 months ago
- A Collection of Papers on Diffusion Language Models☆154Updated 4 months ago
- [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification☆32Updated 10 months ago
- CoT-Valve: Length-Compressible Chain-of-Thought Tuning☆89Updated 11 months ago
- A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention☆274Updated last month
- The most open diffusion language model for code generation — releasing pretraining, evaluation, inference, and checkpoints.☆505Updated 2 months ago
- [ICML'25] Our study systematically investigates massive values in LLMs' attention mechanisms. First, we observe massive values are concen…☆85Updated 7 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆161Updated 3 months ago
- [NeurIPS 2025] Think Silently, Think Fast: Dynamic Latent Compression of LLM Reasoning Chains☆71Updated 6 months ago
- [ICLR 2025] Dynamic Mixture of Experts: An Auto-Tuning Approach for Efficient Transformer Models☆152Updated 6 months ago
- The official repository for the paper "ThinkMorph: Emergent Properties in Multimodal Interleaved Chain-of-Thought Reasoning"☆140Updated 3 weeks ago
- LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification☆73Updated 6 months ago
- SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs☆187Updated 4 months ago
- Efficient triton implementation of Native Sparse Attention.☆261Updated 8 months ago
- Source code for paper "Empirical Analysis of Decoding Biases in Masked Diffusion Models"☆33Updated 3 weeks ago