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
Atualizado 3/12/2026
Estrelas
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.
Instalação
$ install --globalskills.sh
Uso
Depois de instalar, você pode usar esta skill executando o seguinte comando no terminal:
skills use add-pallas-kernel