Search results

Problems with TMA example.

Status: Open.

#348 In jax-ml/jax-triton;

· southfreebird opened on Jul 24, 2025

Can't run Example

Status: Open.

#337 In jax-ml/jax-triton;

· ando600 opened on Mar 29, 2025

Clarify that jax-triton only works on GPU

Status: Open.

#276 In jax-ml/jax-triton;

· skye opened on Apr 18, 2024

Is scan supported in pallas?

Status: Open.

#202 In jax-ml/jax-triton;

· hr0nix opened on Jul 17, 2023

Example notebook with attention isn't working

Status: Open.

#180 In jax-ml/jax-triton;

· hr0nix opened on Jun 24, 2023

Distributed array support?

Status: Open.

#163 In jax-ml/jax-triton;

· sh0416 opened on Jun 7, 2023

Internal bounds-check failed when calling FlashAttention kernel from jax-triton

Status: Open.

#135 In jax-ml/jax-triton;

· hbarclay opened on May 9, 2023

Question about Purpose of Pallas

Status: Open.

#124 In jax-ml/jax-triton;

· adam-hartshorne opened on Apr 20, 2023

Use <code>XlaCustomCallStatus</code> to indicate CUDA launch failures

enhancementNew feature or requestNew feature or request

Status: Open.

#25 In jax-ml/jax-triton;

· sharadmv opened on Nov 3, 2022