skills.homes logoskills.homescapability registry
home/categories/framework-internals/marin-community-marin-agents-skills-add-pallas-kernel-skill-md
cd ..
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.

View Sourceframework-internals
marin-community
maintainer
marin-community
Updated 3/12/2026
Stars
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
Usage

Once installed, you can use this skill by running the following command in your terminal:

skills use add-pallas-kernel