search

jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
354 stars 40 forks source link

Migrated ``pl.BlockSpec`` uses in jax_triton to the new argument order #279

Closed copybara-service[bot] closed 5 months ago

copybara-service[bot] commented 5 months ago

Migrated pl.BlockSpec uses in jax_triton to the new argument order

See google/jax#22209.