Issues

• 32 different nccl channel created in the decode step of DSR1 671b (IREE PJRT as backend)

  jax-ml/jax-llm-examples#50 · Filter by author Hanjiang9831 opened on Apr 26, 2026
  1

• Question: Why is there is a big difference in Token throughput for Deepseek R1?

  jax-ml/jax-llm-examples#43 · Filter by author arul-lm opened on Dec 27, 2025
  2

• Question about loading weights with orbax

  jax-ml/jax-llm-examples#40 · Filter by author yuanfz98 opened on Nov 21, 2025
  1

• Model support for DeepSeekV3.1Terminus and DeepSeekV3.2 Exp

  jax-ml/jax-llm-examples#37 · Filter by author yuanfz98 opened on Oct 10, 2025
  1

• Feature Request For supporting GLM4.5V and Deepseek V3.1

  jax-ml/jax-llm-examples#35 · Filter by author yuanfz98 opened on Sep 11, 2025
  1

• Feature Request For Speculative Decoding

  jax-ml/jax-llm-examples#34 · Filter by author yuanfz98 opened on Sep 10, 2025
  1

• Possibility of reducing KVCache size

  jax-ml/jax-llm-examples#33 · Filter by author yuanfz98 opened on Sep 9, 2025
  1

• Deepseek moe jax Training code

  jax-ml/jax-llm-examples#20 · Filter by author Lisennlp opened on Jul 11, 2025
  1

• Error while running main.py - Most likely missing chkpoint location

  jax-ml/jax-llm-examples#1 · Filter by author shivajid opened on Mar 5, 2025
  1