skills.homes logoskills.homescapability registry
home/categories/framework-internals/jkitchin-skillz-skills-programming-python-jax-skill-md
cd ..
framework-internalsdevelopment

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

سورس دیکھیںframework-internals
jkitchin
maintainer
jkitchin
اپ ڈیٹ ہوا 3/7/2026
اسٹارز
24
فورکس
5
quick start

Installation and usage

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

انسٹالیشن
$ install --globalskills.sh
استعمال

انسٹال کرنے کے بعد، آپ یہ اسکل ٹرمینل میں درج ذیل کمانڈ چلا کر استعمال کر سکتے ہیں:

skills use python-jax