Discussions

• 1 You must be logged in to vote

  🙏

  We found the model outputs like logits are very different when using eager vs flash attention on TPU with the same input, model, etc. except the attention choice

  weirayao asked Sep 6, 2025 in Q&A · Unanswered

  

  

  0
• 1 You must be logged in to vote

  🙏

  Inquiry on Torch XLA Scalability and Performance Comparison with JAX on TPU

  Rayjryang asked Jun 19, 2025 in Q&A · Unanswered

  

  

  

  

  2
• 1 You must be logged in to vote

  🙏

  How to contribute?

  hlnchen asked Jun 2, 2025 in Q&A · Answered

  

  

  2
• 1 You must be logged in to vote

  🙏

  Data seems not sharded across processes in multi-host single-slice setting

  weirayao asked May 31, 2025 in Q&A · Answered

  

  

  3