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

by jkitchin· Repository·development

Run in AgentArea Browse All Skills

Also installable via skills CLI

npx skills add jkitchin/skillz/skills/programming/python-jax

Source

Repo:SkillsMP + GitHub Raw

Path:skills/programming/python-jax(main)

Related in development

fix-facebook-react

Use when you have lint errors, formatting issues, or before committing code to ensure it passes CI.

by facebook

242,476

update-docs-vercel-next-js

This skill should be used when the user asks to "update documentation for my changes", "check docs for this PR", "what docs need u...

by vercel

137,316

docstring-pytorch-pytorch

Write docstrings for PyTorch functions and methods following PyTorch conventions. Use when writing or updating docstrings in PyTor...

by pytorch

96,875