Moved Pallas GPU lowering registartion code into a separate submodule

jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

Apache License 2.0

340 stars 36 forks source link

Closed copybara-service[bot] closed 8 months ago

copybara-service[bot] commented 8 months ago

Moved Pallas GPU lowering registartion code into a separate submodule

This makes the layout similar to the one we use in Pallas TPU.