Agent Skills: jax-skills

High-performance numerical computing and machine learning workflows using JAX. Supports array operations, automatic differentiation, JIT compilation, RNN-style scans, map/reduce operations, and gradient computations. Ideal for scientific computing, ML models, and dynamic array transformations.

UncategorizedID: benchflow-ai/skillsbench/jax-skills

Author

benchflow-ai

https://github.com/benchflow-ai View all skills

Repository

benchflow-ai/skillsbench

benchflow-aiLicense: Apache-2.0

1,819369

Install this agent skill to your local

pnpm dlx add-skill https://github.com/benchflow-ai/skillsbench/jax-skills

Skill Files

Browse the full folder contents for jax-skills.

Download Skill

Loading file tree…

Select a file to preview its contents.