jax-best-practices SKILL.md skillA
jax-best-practices from mindrally/skills is an agent skill: a SKILL.md folder an AI agent loads when a task calls for it.
mdr add mindrally/skills/jax-best-practices@git:20261008.7682ca7Pins git:20261008.7682ca7 in mdr.lock. Pin the fingerprint instead.
- Audit grade
- A, 17 of 17 checks passed
- Kind
- SKILL.md skill
- Repository
- mindrally/skills
- Path
jax-best-practices/SKILL.md- Versions recorded
- 2
- Latest version
- git:20261008.7682ca7, committed Oct 8, 2026
- First version recorded
- Jan 23, 2026
- Size
- 1,456 bytes
- Repository stars
- 278
- License
- Apache-2.0
Expert in JAX for high-performance numerical computing and machine learning.
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
# JAX Best Practices You are an expert in JAX for high-performance numerical computing and machine learning. ## Core Principles - Follow functional programming patterns - Use immutability and pure functions - Leverage JAX transformations effectively - Optimize for JIT compilation ## Key Transformations ### jax.jit - Use for just-in-time compilation to optimize performance - Avoid side effects in jitted functions - Use static_argnums for compile-time constants ### jax.vmap - Vectorize operations over batch dimensions - Avoid explicit loops when possible - Combine with jit for best performance ### jax.grad - Compute gradients automatically - Use for automatic differentiation - Combine with jit for efficient gradient computation ## Best Practices - Write pure functions without side effects - Use JAX arrays instead of NumPy where possible - Leverage random key splitting properly - Profile and optimize hot paths ## Performance - Minimize Python overhead in hot loops - Use appropriate dtypes - Batch operations when possible - Profile with JAX profiler ## Common Patterns - Use pytrees for nested data structures - Implement custom vjp/jvp when needed …
Read the whole file at its exact version.
How to install
mdr add mindrally/skills/jax-best-practices@sha256:1ef7fa7b63b86905The line at the top of this page pins the label git:20261008.7682ca7, 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.
[](https://markdownregistry.com/a/art_sjjznpda6tjbo6d4)
1 badge image fetches in 30 days. What the badge shows.
Versions
The registry has recorded 2 versions of this file, between Jan 23, 2026 and Oct 8, 2026, about one every 258 days.
| version | committed | commit | size | audit | |
|---|---|---|---|---|---|
| git:20261008.7682ca7 latest | 2026-10-08 | 7682ca7 | 1,456 B | A | view · diff |
| git:20260123.47f47c1 | 2026-01-23 | 47f47c1 | 1,376 B | A | view |
Audit of the latest version
- pass: Frontmatter block present
- pass: Frontmatter declares a name
- pass: Frontmatter declares a description
- pass: Size between 200 bytes and 200 KB (1456 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
mindrally/skills · 278 stars · license Apache-2.0 · pushed 2026-10-08 · branch main
API
GET https://markdownregistry.com/api/v1/artifacts/art_sjjznpda6tjbo6d4 GET https://markdownregistry.com/api/v1/resolve?ref=mindrally/skills/jax-best-practices GET https://markdownregistry.com/api/v1/blob/1ef7fa7b63b86905d92efc8b1e8eba60cacd7adb1260aba74c7f2ad8437a53ac
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.