Skill Store 33

Jax Development

Write, debug, review, profile, or shard JAX numerical code.

by tristanmanchester

The skill

Write, debug, review, profile, or shard JAX numerical code. Use when the hard part is JAX tracing, autodiff, control flow, PRNGs, compilation, array placement, or runtime performance. Do not impose JAX on a NumPy-only or unrelated GPU task.

This skill's licence does not allow redistribution, so the store does not show its text. Read it at the source.

Use this skill

Connect your agent to the Skill Store MCP server and ask for it by ref: tristanmanchester/jax-development-10780196969.

Markdown version

Details

Publisher
tristanmanchester
Category
Software Development › Languages & frameworks
Updated
2026-09-14

References

More from tristanmanchester

  • Agent Use Audit or design agent-facing product capabilities: action/context parity, discoverable contracts,…
  • Ai Codebase Deep Modules Design or refactor cohesive modules with small, useful public contracts and enforceable dependenc…
  • Animating React Native Expo Implement and debug React Native or Expo motion with Reanimated 4 and Gesture Handler 3: state tr…
  • Assemblyai Transcribe Transcribe or post-process audio with AssemblyAI when explicitly requested or already selected by…

Related skills

  • Jax Best Practices Expert in JAX for high-performance numerical computing and machine learning
  • Machine Learning Machine learning development with JAX, functional programming patterns, and high-performance comp…
  • Jax Composable transformations of Python+NumPy programs.
  • Java Profiling JVM performance profiling with Java Flight Recorder (JFR), jcmd, and GC analysis.
  • 162 Java Profiling Analyze Use when you need to analyze Java profiling data collected during the detection phase — including…