evo-jax-tasks

evo-jax-tasks is a skill for Claude Code, Codex from Zhang-Henry/CoEvoSkills. It costs 33 tokens per session (270 once invoked), scanned A, original, Apache-2.0.

A task solver for JAX, a Python library for numerical and machine-learning computation. It reads a problem manifest and handles reductions, vectorized mapping, gradients, scans, and just-in-time compiled models.

In plain words
What is it for?
Running tasks such as row means, element-wise squares, logistic-loss gradients, recurrent-network passes, and small neural-network computations.
Why use it?
It provides a repeatable way to solve and check a defined set of JAX programming tasks.

Skill for Claude CodeCodex

Written for no agent in particular: nothing here depends on one.

Good fit Running tasks such as row means, element-wise squares, logistic-loss gradients, recurrent-network passes, and small neural-network computations.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/zhang-henry/coevoskills/evo-jax-tasks
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent
npx skills add Zhang-Henry/CoEvoSkills --skill evo-jax-tasks
Clone the repo
git clone --depth 1 https://github.com/Zhang-Henry/CoEvoSkills

Made for: Claude Code, Codex.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for evo-jax-tasks

README.md
[![agentmods](https://agentmods.dev/badge/skills/zhang-henry/coevoskills/evo-jax-tasks/github.svg)](https://agentmods.dev/skills/zhang-henry/coevoskills/evo-jax-tasks)
Your own site
<a href="https://agentmods.dev/skills/zhang-henry/coevoskills/evo-jax-tasks"><img src="https://agentmods.dev/badge/skills/zhang-henry/coevoskills/evo-jax-tasks/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for evo-jax-tasks

Your own site · 80×15
<a href="https://agentmods.dev/skills/zhang-henry/coevoskills/evo-jax-tasks"><img src="https://agentmods.dev/badge/skills/zhang-henry/coevoskills/evo-jax-tasks.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 33 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 270 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. A grade says what 26 rules found in the file — not that it is safe. Third-party audits
  • NVIDIA SkillSpector pass 7 Sept 2026
How audits are shown
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00033 $0.00270
Opus 5 $0.00016 $0.00135
Sonnet 5 $0.00007 $0.00054
Haiku 4.5 $0.00003 $0.00027

Measured 12d ago against content hash 56028114bd95, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-11, from the pricing page.

Security

Grade A, and why

evo-jax-tasks scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 12d ago.

The scan reads SKILL.md. This mod also ships 2 executable files (scripts/__init__.py, scripts/utils.py), listed below but not scanned — reading those needs a real analyzer, not pattern matching.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

artifacts/skills/jax-computing-basics/evo-jax-tasks/SKILL.md · 39 lines

What it actually says

JAX Task Solver Skill

This skill solves sets of JAX programming tasks specified in a problem.json manifest.

Supported Task Types

  • basic_reduce: Row-wise mean reduction
  • map_square: Element-wise squaring via vmap
  • grad_logistic: Gradient of logistic loss via jax.grad
  • scan_rnn: RNN forward pass via jax.lax.scan
  • jit_mlp: JIT-compiled 2-layer MLP forward pass

Usage

import sys
sys.path.insert(0, '/app/environment/skills/evo-jax-tasks/scripts')
from utils import run_all_tasks, validate_outputs

# Run all tasks
run_all_tasks('/app/problem.json', '/app')

# Validate outputs
validate_outputs('/app/problem.json', '/app')

Individual Task Functions

Each task function takes (input_path, output_path) and returns the JAX result:

from utils import task_basic_reduce, task_map_square, task_grad_logistic, task_scan_rnn, task_jit_mlp
Files

What ships with it

2 files beside SKILL.md in the same directory: the scripts, references and assets a skill reads on demand. Not counted in the per-session cost; read them before you install if any of them is executable.

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 12d ago First seen · 39 lines · 33 tokens per session scan A 56028114bd95

Subscribe to this mod's changes

evo-jax-tasks is a skill published in the GitHub repository Zhang-Henry/CoEvoSkills (66 stars, last pushed 22d ago), licensed Apache-2.0. It adds 33 tokens to every session and 270 once invoked, about $0.0002 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.

Related

Other skills, from other repositories

jupyter-notebook

Iterative Python via live Jupyter kernel (hamelnb).

NousResearch/hermes-agent · 18 tokens

matlab

Build, review, migrate, and safely plan MATLAB or GNU Octave numerical workflows, including arrays, tabular/time data, tests, projects, graphics, MAT files, and explicit Python interoperability.

K-Dense-AI/scientific-agent-skills · 42 tokens

bioservices

Unified Python interface to 40+ bioinformatics services. Use when querying multiple databases (UniProt, KEGG, ChEMBL, Reactome) in a single workflow with consistent API. Best for cross-database analysis, ID mapping across services. For quick single-database lookups use gget; for sequence/file manipulation use…

K-Dense-AI/scientific-agent-skills · 73 tokens

pennylane

Hardware-agnostic quantum ML framework with automatic differentiation. Use when training quantum circuits via gradients, building hybrid quantum-classical models, or needing device portability across IBM/Google/Rigetti/IonQ. Best for variational algorithms (VQE, QAOA), quantum neural networks, and integration with…

K-Dense-AI/scientific-agent-skills · 98 tokens

cuopt-numerical-optimization-api

LP, MILP, and QP (beta) with cuOpt — Python, C, and CLI. Use when the user is solving LP, MILP, or QP with any cuOpt interface.

NVIDIA/skills · 51 tokens

rocm-kernels

Provides guidance for writing and benchmarking optimized Triton kernels for AMD GPUs (MI355X, R9700) on ROCm, targeting HuggingFace diffusers (LTX-Video, SD3, FLUX) and transformers. Core kernels: RMSNorm, RoPE 3D, GEGLU, AdaLN. Includes XCD swizzle, autotune, diffusers integration patterns, and LTX-Video pipeline…

huggingface/kernels · 93 tokens