framework-internalsdevelopment
add-pallas-kernel
Add or update a TPU kernel using jax.experimental.pallas. Use when asked to implement, modify, benchmark, or autotune a Pallas TPU/GPU kernel.
maintainer
marin-community
Actualizado 3/12/2026
Estrellas
844
Forks
103
quick start
Installation and usage
Add or update a TPU kernel using jax.experimental.pallas. Use when asked to implement, modify, benchmark, or autotune a Pallas TPU/GPU kernel.
Instalación
$ install --globalskills.sh
Uso
Después de instalarlo, puedes usar este skill ejecutando el siguiente comando en tu terminal:
skills use add-pallas-kernel