Alternatives and similar repositories for sglang-jax

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

• Dao-AILab / sonic-moe
  Accelerating MoE with IO and Tile-aware Optimizations
  ☆563Updated last week
• ByteDance-Seed / ByteCheckpoint
  ByteCheckpoint: An Unified Checkpointing Library for LFMs
  ☆264Updated last month
• gpu-mode / ring-attention
  ring-attention experiments
  ☆163Updated last year
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆289Updated 4 months ago
• vllm-project / tpu-inference
  TPU inference for vLLM, with unified JAX and PyTorch support.
  ☆223Updated this week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆277Updated this week
• meta-pytorch / KernelAgent
  Autonomous GPU Kernel Generation via Deep Agents
  ☆223Updated this week
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆175Updated 9 months ago
• fzyzcjy / torch_memory_saver
  Allow torch tensor memory to be released and resumed later
  ☆207Updated 2 weeks ago
• meta-pytorch / torchcomms
  torchcomms: a modern PyTorch communications API
  ☆323Updated this week
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆314Updated 5 months ago
• meta-pytorch / BackendBench
  Ship correct and fast LLM kernels to PyTorch
  ☆139Updated 2 weeks 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…
  ☆252Updated last week
• fzyzcjy / torch_utils
  Utility scripts for PyTorch (e.g. Make Perfetto show some disappearing kernels, Memory profiler that understands more low-level allocatio…
  ☆81Updated 4 months ago
• HazyResearch / Megakernels
  kernels, of the mega variety
  ☆657Updated 4 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆93Updated 9 months ago
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆262Updated 3 months ago
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆319Updated this week
• triton-lang / kernels
  ☆101Updated last year
• tspeterkim / paged-attention-minimal
  a minimal cache manager for PagedAttention, on top of llama3.
  ☆133Updated last year
• perplexityai / pplx-kernels
  Perplexity GPU Kernels
  ☆554Updated 2 months ago
• osayamenja / FlashMoE
  Distributed MoE in a Single Kernel [NeurIPS '25]
  ☆188Updated this week
• hao-ai-lab / cse234-w25-PA
  ☆45Updated 10 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆141Updated 8 months ago
• meta-pytorch / kraken
  Triton-based Symmetric Memory operators and examples
  ☆80Updated 2 weeks ago
• dropbox / gemlite
  Fast low-bit matmul kernels in Triton
  ☆424Updated this week
• sail-sg / zero-bubble-pipeline-parallelism
  Zero Bubble Pipeline Parallelism
  ☆448Updated 8 months ago
• NVIDIA / nvshmem
  NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…
  ☆459Updated 3 weeks ago
• MDK8888 / vllmini
  A minimal implementation of vllm.
  ☆66Updated last year
• deepseek-ai / LPLB
  An early research stage expert-parallel load balancer for MoE models based on linear programming.
  ☆491Updated 2 months ago