python-jax

Expert guidance for JAX (Just After eXecution) - high-performance numerical computing with automatic differentiation, JIT compilation, vectorization, and GPU/TPU acceleration; includes transformations (grad, jit, vmap, pmap), sharp bits, gotchas, and differences from NumPy

$ インストール

git clone https://github.com/jkitchin/skillz /tmp/skillz && cp -r /tmp/skillz/skills/programming/python-jax ~/.claude/skills/skillz

// tip: Run this command in your terminal to install the skill