Skip to content
Agentmart
Submit a tool

jax-skills

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

Skill by benchflow-ai

IndexedProprietary. LICENSE.txt has complete terms★ 1,870
Write a review

Install

Works with Claude Code, Claude apps, Codex, Cursor, Gemini CLI, GitHub Copilot, VS Code, OpenCode, Goose, Windsurf. Read the repository before you give any tool access to your data.

npx agentmarket install skill/benchflow-ai-jax-skills

Asks which of your tools to add it to (Claude Code, Codex, Cursor, VS Code, Windsurf, Gemini CLI, Claude Desktop) and shows the change first. See https://agentmart.dev/cli.

Get the weekly diff

The weekly diff: one email a week with new agent tools, notable updates and grade changes, plus security alerts for tools you follow. Double opt-in, one-click unsubscribe.

We use your email only for this newsletter. Privacy

About

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.

Use cases: ml, research

Reviews

Write a review

No reviews yet. Installed it? Reviews come from verified installers only. How reviews work

Permissions

What this skill can do and what it asks for, from its SKILL.md. Checked Oct 25, 2026.

Tools
No tools found
Secrets
None asked for
Install scripts
None found
Runs as
Python package

Network hosts it talks to 1

  • codeload.github.com · found in install command

Files it touches

  • ~/.claude/skills/jax-skills

What's inside

Skill name
jax-skills
Description
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.
License
Proprietary. LICENSE.txt has complete terms
Instructions
148 lines in SKILL.md · view on GitHub

Before installing

  • Ships executable scripts

Files (9)

  • 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

The license isn't open source, so the instructions aren't shown here. Read SKILL.md on GitHub

More from benchflow-ai/skillsbench

Signals

Stars
1,870
Forks
372
Open issues
50
Last commit (this folder)
Jan 19, 2026 (9 months ago)
Last release
Jun 14, 2026 (4 months ago)

From GitHub (fetched Oct 24, 2026).

Details

License
Proprietary. LICENSE.txt has complete terms
Repository
github.com/benchflow-ai/skillsbench
Website
www.skillsbench.ai
First seen
Oct 24, 2026
Last checked
Oct 24, 2026
Source
GitHubgithub-graphql

More from benchflow-ai

Cite this page

jax-skills. Agent Mart. https://agentmart.dev/skills/benchflow-ai-jax-skills (last updated 2026-10-24)