Removed Pallas forwarding modules from jax_triton

jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

Apache License 2.0

354 stars 40 forks source link

Closed copybara-service[bot] closed 7 months ago

copybara-service[bot] commented 7 months ago

Removed Pallas forwarding modules from jax_triton

The extra layer of indirection is unnecessary.