Alternatives and similar repositories for sglang-jax

Users that are interested in sglang-jax are comparing it to the libraries listed below

• ByteDance-Seed / ByteCheckpoint
  ByteCheckpoint: An Unified Checkpointing Library for LFMs
  ☆249Updated 3 months ago
• sgl-project / genai-bench
  Genai-bench is a powerful benchmark tool designed for comprehensive token-level performance evaluation of large language model (LLM) serv…
  ☆222Updated last week
• fzyzcjy / torch_memory_saver
  Allow torch tensor memory to be released and resumed later
  ☆157Updated last week
• fzyzcjy / torch_utils
  Utility scripts for PyTorch (e.g. Memory profiler that understands more low-level allocations such as NCCL)
  ☆62Updated last month
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆125Updated 5 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆72Updated 5 months ago
• meta-pytorch / kraken
  Triton-based Symmetric Memory operators and examples
  ☆48Updated 2 weeks ago
• InternLM / turbomind
  ☆97Updated 7 months ago
• yanring / Megatron-MoE-ModelZoo
  Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.
  ☆114Updated 2 weeks ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆155Updated last year
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆216Updated last year
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆123Updated last week
• RLsys-Foundation / TritonForge
  🔥 LLM-powered GPU kernel synthesis: Train models to convert PyTorch ops into optimized Triton kernels via SFT+RL. Multi-turn compilation…
  ☆89Updated 3 weeks ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆60Updated last year
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆263Updated last month
• Infini-AI-Lab / MagicDec
  [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding
  ☆130Updated 10 months ago
• triton-lang / kernels
  ☆93Updated 11 months ago
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆182Updated 3 weeks ago
• flashinfer-ai / cutlass-viz
  ☆65Updated 6 months ago
• sgl-project / sgl-cookbook
  Make SGLang go brrr
  ☆37Updated last month
• fanshiqing / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆156Updated 3 weeks ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆242Updated last week
• Victarry / PP-Schedule-Visualization
  Pipeline Parallelism Emulation and Visualization
  ☆69Updated 4 months ago
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆301Updated 2 months ago
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆148Updated 2 weeks ago
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆272Updated this week
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆80Updated 11 months ago
• yifuwang / symm-mem-recipes
  ☆144Updated 10 months ago
• stepfun-ai / StepMesh
  ☆309Updated last month
• cchan / tccl
  extensible collectives library in triton
  ☆90Updated 7 months ago