Skills
.Work. You
Rest

jax-equinox-numerics

Community

JAX+Equinox numerics, standardized.

Software Engineering#patterns#jax#jit#equinox#numerics#pytrees#ad
Authorquattro
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This playbook distills and standardizes JAX + Equinox numerics patterns into a repo-agnostic set of rules, checklists, and practical guidance to improve reliability, stability, and performance in scientific code.

Core Features & Use Cases

  • Repo-agnostic patterns distilled from Equinox, Lineax, Optimistix, and Diffrax for broad applicability.
  • Actionable checklists covering JIT boundaries, PyTree handling, and numerical stability.
  • Guided adoption for teams seeking to raise engineering discipline in numerical software.

Quick Start

Apply these best practices to your JAX/Equinox numerics project to standardize patterns.

Dependency Matrix

Required Modules

None required

Components

Standard package

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: jax-equinox-numerics
Download link: https://github.com/quattro/jax-numerics-agent/archive/main.zip#jax-equinox-numerics

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 223,000+ vetted skills library on demand.