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
Mis à jour 3/12/2026
Étoiles
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.
Installation
$ install --globalskills.sh
Utilisation
Après l'installation, vous pouvez utiliser ce skill en exécutant la commande suivante dans votre terminal :
skills use add-pallas-kernel