cuequivariance-jax SKILL.md skillA

This file is byte-identical to the first copy the registry indexed, one of 3 indexed copies. Same content hash, so the same checks give both the same grade.

cuequivariance-jax from nvidia-bionemo/bionemo-agent-toolkit is an agent skill: a SKILL.md folder an AI agent loads when a task calls for it.

Install it, pinned
mdr add nvidia-bionemo/bionemo-agent-toolkit/cuequivariance-jax@git:20261105.dbff972

Pins git:20261105.dbff972 in mdr.lock. Pin the fingerprint instead.

Audit grade
A, 17 of 17 checks passed
Kind
SKILL.md skill
Repository
nvidia-bionemo/bionemo-agent-toolkit
Path
library-skills/cuequivariance-jax/SKILL.md
Versions recorded
1
Latest version
git:20261105.dbff972, committed Nov 5, 2026
Size
19,352 bytes
Repository stars
503
License
a license file GitHub does not identify

Use when writing or debugging cuEquivariance JAX polynomials, irrep dictionaries, Flax NNX layers, or CPU/CUDA execution and layouts.

Collected from public GitHub. Every version is saved exactly as it was and never changes. Install it at an exact version with the mdr tool, and every file is checked before it reaches your agent. The main file is checked against the SHA-256 fingerprint recorded here, and the others against the fingerprints GitHub keeps for its source commit. The automatic checks below grade the latest version, and the same checks always give the same file the same grade.

What the file says

# cuEquivariance in JAX

## Purpose

Execute `cuequivariance` (`cue`) descriptors with `cuequivariance_jax` (`cuex`). Preserve the project's framework and devices. Custom representations use core `cuequivariance`; PyTorch modules use `cuequivariance_torch`.

## Inputs

Required inputs: operation, SO3/O3 irreps, layout, batch shapes, dtype, device, and weight ownership. Derive them from the request/code; state defaults. Graphs also need indices and destination count. Operand sizes/output irreps come from descriptors. Optional inputs: backend/performance preferences.

## Instructions

1. Probe the project interpreter once and reuse working packages. **Environment setup/recovery must not recursively delete directories**, including temporary environments. Use a fresh unique path when isolation is needed.
2. Select `method="naive"` for CPU; Flax is needed only for NNX. High-level dictionary blocks are `(..., mul, ir.dim)`; uniform CUDA arrays differ.
3. Adapt one matching recipe into a script. Build descriptors outside transformed functions, broadcast leading input axes, and enable x64 **before array creation** for float64 checks.
…

Read the whole file at its exact version.

How to install

The exact contents, by fingerprint
mdr add nvidia-bionemo/bionemo-agent-toolkit/cuequivariance-jax@sha256:59273471279715c2

The line at the top of this page pins the label git:20261105.dbff972, which is what most people want. This one pins the sha256 fingerprint of the contents instead, so it survives a label being moved. Either way the exact version is written to mdr.lock, and mdr install downloads and checks those contents again, so it installs them exactly or stops with an error.

Badge

mdr badge

[![mdr](https://markdownregistry.com/badge/art_yv53fph6czasjglj.svg)](https://markdownregistry.com/a/art_yv53fph6czasjglj)

1 badge image fetches in 30 days. What the badge shows.

Versions

The registry has recorded 1 version of this file, committed on Nov 5, 2026.

versioncommittedcommitsizeaudit
git:20261105.dbff972 latest2026-11-05 dbff972 19,352 BA view

Audit of the latest version

A  17 of 17 checks passed. Fixed automatic checks with no AI involved, so the same checks give the same answer every run.
  • pass: Frontmatter block present
  • pass: Frontmatter declares a name
  • pass: Frontmatter declares a description
  • pass: Size between 200 bytes and 200 KB (19352 bytes)
  • pass: No zero-width or bidi control characters
  • pass: No instruction hidden inside an HTML comment
  • pass: No link to an exfiltration or paste host
  • pass: No credential-shaped string
  • pass: No instruction to send local credentials anywhere
  • pass: No text hidden with inline styles
  • pass: No prompt-injection phrasing
  • pass: No curl or wget piped into a shell
  • pass: No recursive delete of root, home or parent
  • pass: No instruction to read or print local credentials
  • pass: No base64 blob over 200 characters
  • pass: No link to a raw IP address
  • pass: No script tag

Source

GitHub

nvidia-bionemo/bionemo-agent-toolkit · 503 stars · license a license file GitHub does not identify · pushed 2026-11-05 · branch main

API

GET https://markdownregistry.com/api/v1/artifacts/art_yv53fph6czasjglj
GET https://markdownregistry.com/api/v1/resolve?ref=nvidia-bionemo/bionemo-agent-toolkit/cuequivariance-jax
GET https://markdownregistry.com/api/v1/blob/59273471279715c22098fe71801d3abd0ecc8dadc294cc960689d1777596180c

Your AI assistant does the legwork. You hear about the deals worth your word. Hand yours the standing instructions at modelranch.com and it joins the network that reads files like this one.

Guides: the SKILL.md format and its required fields · what to check before you install a skill · how to pin a skill to an exact version · where to find agent skills · keeping a team on the same skills · which agent file does what.

More from nvidia-bionemo/bionemo-agent-toolkit

cuequivariance-torch SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · library-skills/cuequivariance-torch/SKILL.md · Use when building or debugging cuEquivariance PyTorch tensor products or equivariant GNNs, especially CPU references…
git:20261105.dbff972 · audit A · 503 stars
cuequivariance SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · library-skills/cuequivariance/SKILL.md · Use when defining cuEquivariance groups and irreps, building equivariant tensor products, or validating segmented…
git:20261105.dbff972 · audit A · 503 stars
genomics-workflow-acceleration SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · library-skills/genomics-workflow-acceleration/SKILL.md · Use when accelerating existing genomics workflows with NVIDIA Parabricks, improving runtime or price/performance…
v1.1.0 · audit A · 503 stars
nvmolkit-usage SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · library-skills/nvmolkit-usage/SKILL.md · Use when writing or debugging nvMolKit Python code for GPU-accelerated RDKit fingerprints, similarity, conformers…
git:20261016.a6eced5 · audit A · 503 stars
parabricks SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · library-skills/parabricks/SKILL.md · Plan NVIDIA Parabricks pbrun commands, translate CPU options and check runtime readiness. Use for individual tools…
v1.4.0 · audit A · 503 stars
boltz2-nim SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · nim-skills/boltz2-nim/SKILL.md · Use Boltz2 NIM for biomolecular structure prediction and binding affinity. Invoke for Boltz2, protein structures…
git:20261001.237ee05 · audit A · 503 stars
diffdock-nim SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · nim-skills/diffdock-nim/SKILL.md · Run DiffDock molecular docking via NVIDIA NIM to predict small-molecule binding poses against protein targets. Use for…
git:20260923.5b3088e · audit A · 503 stars
evo2-nim SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · nim-skills/evo2-nim/SKILL.md · Generate and analyze DNA sequences using NVIDIA's Evo 2 BioNeMo NIM microservice. Use for Evo2/Evo 2, DNA generation…
git:20260929.93d4978 · audit A · 503 stars
genmol-nim SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · nim-skills/genmol-nim/SKILL.md · Generate novel drug-like molecules using the GenMol NIM microservice. Use for de novo generation, scaffold decoration…
git:20260930.59b46bd · audit A · 503 stars
drug-discovery-pipeline SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · nim-skills/meta-skills/drug-discovery-pipeline/SKILL.md · NOTE: molecule and target inputs and your NGC_API_KEY are transmitted to external NVIDIA-hosted API endpoints on every…
git:20260812.47315cc · audit A · 503 stars
msa-structure-prediction-pipeline SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · nim-skills/meta-skills/msa-structure-prediction-pipeline/SKILL.md · NOTE: your protein sequence and the retrieved MSA alignment are transmitted to external NVIDIA-hosted APIs…
git:20260812.47315cc · audit A · 503 stars
molmim-nim SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · nim-skills/molmim-nim/SKILL.md · Use this skill for MolMIM, NVIDIA's BioNeMo NIM microservice for small-molecule latent-space generation and…
git:20260923.b36e938 · audit A · 503 stars

Every file in nvidia-bionemo/bionemo-agent-toolkit

Other files named cuequivariance-jax

cuequivariance-jax SKILL.md skill
nvidia/skills · skills/bionemo-cuequivariance-jax/SKILL.md · Use when writing or debugging cuEquivariance JAX polynomials, irrep dictionaries, Flax NNX layers, or CPU/CUDA…
git:20261105.00ce147 · audit A · 3,871 stars
cuequivariance-jax SKILL.md skill
nvidia-bionemo/bionemo-agent-toolkit · skills/bionemo-agent-toolkit/skills/cuequivariance-jax/SKILL.md · Use when writing or debugging cuEquivariance JAX polynomials, irrep dictionaries, Flax NNX layers, or CPU/CUDA…
git:20261105.dbff972 · audit A · 503 stars

Browse by kind, by grade A, or by owner.