search

jax-ml / jax-triton

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

Pallas now uses MLIR Python builders to lower to Triton IR #249

Closed copybara-service[bot] closed 8 months ago

copybara-service[bot] commented 8 months ago

Pallas now uses MLIR Python builders to lower to Triton IR

This allows us to drop a dependency on the Triton Python package in the future, and delegate ->ptx compilation to XLA.