Agent skill

Jax Skills

by benchflow-ai in benchflow-ai/skillsbench

High-performance numerical computing and machine learning workflows using JAX.

ProprietaryAuto-check passedAI & LLM Engineering

Install Jax Skills

skills CLI
$ npx skills add benchflow-ai/skillsbench --skill jax-skills -a claude-code

Project install by default; add -g for ~/.claude/skills/.

GitHub CLI
$ gh skill install benchflow-ai/skillsbench jax-skills --agent claude-code

Project scope by default; add --scope user for a personal install. Needs GitHub CLI 2.90.0 or later (public preview).

Manual copy
$ git clone --depth 1 https://github.com/benchflow-ai/skillsbench.git skills-src && mkdir -p .claude/skills && cp -r skills-src/tasks/jax-computing-basics/environment/skills/jax-skills .claude/skills/jax-skills && rm -rf skills-src

Use ~/.claude/skills/ instead of .claude/skills for a personal install. The folder must contain SKILL.md.

Claude Code skills documentation · loads skills from .claude/skills/

Facts

Skill name
jax-skills
GitHub stars
1.8k
Token cost
~1.1k tokens
SKILL.md length
386 words
Files
9
Skills in repo
189
Repo updated
First seen
Licence
Proprietary

At a glance

High-performance numerical computing and machine learning workflows using JAX.

  • Works in 5 steps: Loading and Saving Arrays → Map and Reduce Operations → Gradients and Optimization → …
  • Tasks that involve Deep learning
  • SKILL.md covers General Guidelines, 1. Loading and Saving Arrays, 2. Map and Reduce Operations and 3. Gradients and Optimization, plus 2 more sections
  • Runs Python scripts from its folder

What it does

Jax Skills is an agent skill from benchflow-ai/skillsbench. 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.

Its SKILL.md is about 1.1k tokens, which your agent loads only when the skill is triggered. The skill folder holds 8 other files (for example `jax_skills.py`).

It sits in AI & LLM Engineering, covering Deep learning and Machine learning. The repository describes itself as: SkillsBench evaluates how well skills work and how effective agents are at using them.

When your agent uses it

  • Tasks that involve Deep learning
  • Tasks that involve Machine learning

Example prompts

  • “/jax-skills”

Requirements

  • Python 3

Workflow steps

5 steps, taken from the step headings in SKILL.md.

  1. Loading and Saving Arrays
  2. Map and Reduce Operations
  3. Gradients and Optimization
  4. Recurrent Scan
  5. JIT Compilation

What it can do on your machine

Read from SKILL.md and the folder at commit 9a1f4dd. It shows what the files ask for, not the result of running them.

  • Tool permissions

    Pre-approves nothing: there is no allowed-tools line, so your agent's usual permission prompts apply.

    From allowed-tools in the SKILL.md frontmatter.

  • Runs code

    Ships script files (Python), which the agent can run.

    From the folder's file list and the shell code blocks in SKILL.md.

  • Network

    No URLs in SKILL.md.

    From URLs in SKILL.md, links to its own repository left out.

  • Credentials

    Names no API keys, tokens, secrets or passwords.

    From names ending in _API_KEY, _TOKEN, _SECRET, _KEY or _PASSWORD in SKILL.md.

Context cost

Jax Skills loads about 1.1k tokens when it runs. Until then it costs about 76 tokens; SKILL.md has 386 words of instructions outside code blocks.

Always · name and description, kept in context so the agent knows when to use it
~76
When it runs · the whole SKILL.md, loaded when a task matches
~1.1k

Estimates: characters ÷ 4, the usual rule of thumb; real counts depend on the model's tokenizer. Scripts and assets cost tokens only if the agent reads them.

Safety

Auto-check passed

The automated check found no risky patterns in SKILL.md.

Automated static check — not a guarantee. Review scripts before installing. It scans the text of SKILL.md for risky patterns (piping downloads into a shell, reading credential files, hidden Unicode, destructive commands); files beside SKILL.md are not scanned.

SKILL.md

Its licence (Proprietary) doesn't allow us to republish the file, so here is its outline and opening line. It has 386 words (~1,051 tokens).

“Description: Load a JAX-compatible array from a file. Supports .npy and .npz. Parameters:”

— opening of SKILL.md by benchflow-ai, Proprietary
name
jax-skills
license
Proprietary. LICENSE.txt has complete terms

Read the full SKILL.md on GitHub

Files

SKILL.md and 8 other files in tasks/jax-computing-basics/environment/skills/jax-skills of benchflow-ai/skillsbench.

  • SKILL.md
  • jax_grad.skill
  • jax_jit.skill
  • jax_load.skill
  • jax_map.skill
  • jax_reduce.skill
  • jax_save.skill
  • jax_scan.skill
  • jax_skills.py

Open the folder on GitHubat commit 9a1f4dd

Compare with similar skills

Jax Skills next to the 5 skills that share the most tags, products or categories with it. Stars are the repository's; “used in” counts other GitHub owners with a copy.

Jax Skills compared with similar skills
SkillStarsUsed inTokensAuto-checkLicenceRepo updated
Jax Skills this skillbenchflow-ai/skillsbench1.8k—~1.1kAutomated safety check: PassProprietary
Scaffold Examplecomet-ml/comet-examples174—~1kAutomated safety check: PassNone
Christopher ManningK-Dense-AI/mimeo282—~1.8kAutomated safety check: PassMIT
Gpt2 CodegolflazyFrogLOL/Harness_Engineering128—~1.8kAutomated safety check: PassNone
Ian GoodfellowK-Dense-AI/mimeo282—~1.7kAutomated safety check: PassMIT
Ilya SutskeverK-Dense-AI/mimeo282—~1.7kAutomated safety check: PassMIT

Similar skills

  • Scaffold Example

    comet-ml/comet-examples

    Scaffold a brand-new Comet example in this repo from the canonical template under templates/integration-example/.

    174 GitHub stars~1k tokensUpdated 1 mo ago
    AI & LLM EngineeringAuto-check passed
  • Christopher Manning

    K-Dense-AI/mimeo

    Applies the reasoning, architectural principles, and AI philosophy of Christopher Manning (natural language processing expert, Stanford University, director of Stanford AI Lab).

    282 GitHub stars~1.8k tokensUpdated 1 mo ago
    AI & LLM EngineeringAuto-check passed
  • Gpt2 Codegolf

    lazyFrogLOL/Harness_Engineering

    Guidance for implementing neural network inference (like GPT-2) under extreme code size constraints.

    128 GitHub stars~1.8k tokensUpdated 4 mo ago
    AI & LLM EngineeringAuto-check passed
  • Ian Goodfellow

    K-Dense-AI/mimeo

    A skill your agent uses when reasoning about generative AI, adversarial machine learning, neural network security, algorithmic fairness, or deep learning fundamentals.

    282 GitHub stars~1.7k tokensUpdated 1 mo ago
    AI & LLM EngineeringAuto-check passed
  • Ilya Sutskever

    K-Dense-AI/mimeo

    Applies the reasoning style of Ilya Sutskever (deep learning pioneer, co-founder of OpenAI and Safe Superintelligence Inc.) to problems involving AI architecture, scaling laws, alignment, and…

    282 GitHub stars~1.7k tokensUpdated 1 mo ago
    AI & LLM EngineeringAuto-check passed
  • Databricks ML Training

    databricks/databricks-agent-skills

    Official

    Train ML models on Databricks. An agent skill from databricks/databricks-agent-skills.

    345 GitHub stars~4.6k tokensUpdated yesterday
    AI & LLM EngineeringAuto-check passed

More from benchflow-ai/skillsbench

All 189 skills in this repo
  • Lean4 Memories

    benchflow-ai/skillsbench

    This skill should be used when working on Lean 4 formalization projects to maintain persistent memory of successful proof patterns, failed approaches, project conventions, and user preferences…

    1.8k GitHub stars~3.2k tokensUpdated 2 mo ago
    Auto-check passed
  • Senior Data Engineer

    benchflow-ai/skillsbench

    World-class data engineering skill for building scalable data pipelines, ETL/ELT systems, real-time streaming, and data infrastructure.

    1.8k GitHub stars~5.9k tokensUpdated 2 mo ago
    Auto-check passed
  • Ac Branch Pi Model

    benchflow-ai/skillsbench

    AC branch pi-model power flow equations (P/Q and |S|) with transformer tap ratio and phase shift, matching acopf-math-model.md and MATPOWER branch fields.

    1.8k GitHub stars~1.1k tokensUpdated 2 mo ago
    Auto-check passed
  • Civ6lib

    benchflow-ai/skillsbench

    Civilization 6 district mechanics library. An agent skill from benchflow-ai/skillsbench.

    1.8k GitHub stars~1.7k tokensUpdated 2 mo ago
    Auto-check passed
  • D3 Visualization

    benchflow-ai/skillsbench

    Build deterministic, verifiable data visualizations with D3.js (v6).

    1.8k GitHub stars~1.5k tokensUpdated 2 mo ago
    Auto-check passed
  • Dc Power Flow

    benchflow-ai/skillsbench

    DC power flow analysis for power systems. An agent skill from benchflow-ai/skillsbench.

    1.8k GitHub stars~717 tokensUpdated 2 mo ago
    Auto-check passed

Questions about Jax Skills

What does Jax Skills do?

High-performance numerical computing and machine learning workflows using JAX. Jax Skills is an agent skill from benchflow-ai/skillsbench. High-performance numerical computing and machine learning workflows using JAX.

When should I use Jax Skills?

Jax Skills fits situations like: tasks that involve Deep learning; tasks that involve Machine learning.

How do I install Jax Skills in Claude Code?

Run `npx skills add benchflow-ai/skillsbench --skill jax-skills -a claude-code`. Or copy the skill folder (tasks/jax-computing-basics/environment/skills/jax-skills in benchflow-ai/skillsbench) into .claude/skills/jax-skills in your project. Claude Code loads it when a task matches its description.

How do I install Jax Skills in Codex?

Run `npx skills add benchflow-ai/skillsbench --skill jax-skills -a codex`. Or copy the skill folder (tasks/jax-computing-basics/environment/skills/jax-skills in benchflow-ai/skillsbench) into .agents/skills/jax-skills in your project. Codex loads it when a task matches its description.

Can I use Jax Skills in Cursor, Gemini CLI or GitHub Copilot?

Cursor, Gemini CLI, GitHub Copilot and OpenCode also load SKILL.md folders. With the skills CLI, run `npx skills add benchflow-ai/skillsbench --skill jax-skills -a cursor` (or -a gemini-cli, github-copilot or opencode for the others). To copy it by hand, put the folder in .cursor/skills/jax-skills, .gemini/skills/jax-skills, .github/skills/jax-skills and .opencode/skills/jax-skills in your project.

What does Jax Skills need to run?

Going by SKILL.md and its folder, Jax Skills needs Python for the scripts in its folder. Our summary lists: Python 3.

Does Jax Skills access the network?

SKILL.md contains no URLs. Any network use would come from the scripts or tools the agent runs. This is read from the text; nothing was executed.

Is Jax Skills safe to install?

Our automated static check of SKILL.md found no risky patterns, such as piping downloads into a shell, reading credential files or hidden Unicode. It is not a guarantee. Review the folder before installing.

What licence does Jax Skills use?

Jax Skills carries a proprietary licence (declared in SKILL.md). It is published on GitHub, but it is not open source: read the licence file before using or sharing it.

How many tokens does Jax Skills use?

About 1.1k tokens (SKILL.md is roughly 4.2k characters). Agents keep only the skill's name and description in context until a task matches; then they load SKILL.md in full.

What are the alternatives to Jax Skills?

Skills that share tags, products or a category with Jax Skills: Scaffold Example (comet-ml/comet-examples, 174 stars), Christopher Manning (K-Dense-AI/mimeo, 282 stars), Gpt2 Codegolf (lazyFrogLOL/Harness_Engineering, 128 stars) and Ian Goodfellow (K-Dense-AI/mimeo, 282 stars). The comparison table on this page puts their stars, adoption, token cost, safety result and licence side by side.

Who maintains Jax Skills?

benchflow-ai (a GitHub organization) maintains it in benchflow-ai/skillsbench, which has 1,835 GitHub stars. The repository holds 189 skills in this directory. The repository was last updated on July 23, 2026.

Source: benchflow-ai/skillsbench on GitHub. Facts on this page come from the repository at the commit we read; the author's words are quoted as theirs.