using-jax
Applies JAX patterns for scientific Python development. Use when working with JAX, distrax, numpyro, blackjax, or scientific computing. Covers vmap, JIT, RNG handling.
Also installable via skills CLI
npx skills add yallup/claude-skills/development/using-jax