search

jax-ml / jax-triton

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

Nuked Pallas examples from jax_triton #307

Closed superbobry closed 1 month ago

superbobry commented 1 month ago

Pallas is now part of JAX and has its own examples in https://github.com/jax-ml/jax/tree/main/jax/experimental/pallas/ops/gpu.