jax Agent Skills & AI Rules

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

36,154 stars

Python 1 file · ~11,476 tokens #jax

RAW Rules

1 Discovered Rules • ~7 Estimated Tokens

Loading skill content...

oneapi Documentation

jax_plugins/oneapi/README.md plugin-manifest ~7 tokens

Raw

JAX Plugin for Intel GPUs