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:
- Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…☆197Updated 2 months ago
- ☆142Updated 2 weeks ago
- paper list, tutorial, and nano code snippet for Diffusion Large Language Models.☆153Updated 2 weeks ago
- [ICLR 2026] TraceRL & TraDo-8B: Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models☆419Updated last week
- ☆119Updated 4 months ago
- Official PyTorch implementation of the paper "Accelerating Diffusion Large Language Models with SlowFast Sampling: The Three Golden Princ…☆40Updated 6 months ago
- [ICLR 2025] Dynamic Mixture of Experts: An Auto-Tuning Approach for Efficient Transformer Models☆153Updated 6 months ago
- A Collection of Papers on Diffusion Language Models☆155Updated 4 months ago
- Source code for paper "Empirical Analysis of Decoding Biases in Masked Diffusion Models"☆36Updated 3 weeks ago
- ☆64Updated 6 months ago
- CoT-Valve: Length-Compressible Chain-of-Thought Tuning☆89Updated 11 months ago
- A collection of papers on discrete diffusion models☆168Updated 7 months ago
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆289Updated 3 months ago
- PoC for "SpecReason: Fast and Accurate Inference-Time Compute via Speculative Reasoning" [NeurIPS '25]☆61Updated 4 months ago
- ☆55Updated 8 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
- [TMLR 2025] Efficient Reasoning Models: A Survey☆298Updated this week
- Discrete Diffusion Forcing (D2F): dLLMs Can Do Faster-Than-AR Inference☆238Updated 3 weeks ago
- ☆221Updated 2 months ago
- [ICLR 2026] dParallel: Learnable Parallel Decoding for dLLMs☆58Updated last week
- The official implementation of dLLM-Var☆31Updated 3 months ago
- ☆129Updated 8 months ago
- Spectral Sphere Optimizer☆94Updated 3 weeks ago
- [NeurIPS'25] dKV-Cache: The Cache for Diffusion Language Models☆129Updated 8 months ago
- [ICLR 2026] The official repository for paper "ThinkMorph: Emergent Properties in Multimodal Interleaved Chain-of-Thought Reasoning"☆142Updated last week
- A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention☆278Updated 2 months ago
- [ASPLOS'26] Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Drafter☆131Updated 2 months ago
- LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification☆73Updated 6 months ago
- [NeurIPS 2025] Think Silently, Think Fast: Dynamic Latent Compression of LLM Reasoning Chains☆74Updated 6 months ago
- qwen-nsa☆87Updated 3 months ago