search

jax-ml / jax-triton

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

Add JAX→Triton fp8 type mapping. #291

Closed copybara-service[bot] closed 2 months ago

copybara-service[bot] commented 2 months ago

Add JAX→Triton fp8 type mapping.

Both Triton and JAX support them now.