multi-node-jax-ffi-debug
CommunityDebug multi-node JAX FFI on SLURM clusters.
AuthorKangOxford
Version1.0.0
Installs0
System Documentation
What problem does it solve?
Systematically diagnose and resolve multi-node JAX + CUDA FFI kernel failures when using shard_map on SLURM clusters.
Core Features & Use Cases
- Provides a structured six-step fix sequence to diagnose and repair common multi-node JAX FFI issues.
- Guides reliable environment propagation, explicit GPU allocation, distributed initialization, and all-rank synchronization to prevent deadlocks.
- Includes practical coordinator resolution strategies using scontrol and robust wrapper scripts to ensure correct per-node setups.
Quick Start
Follow the six-step checklist to stabilize a multi-node JAX FFI run on SLURM by wrapping environment setup, requesting GPUs, enabling all ranks in distributed init, ensuring collective ops, and resolving the coordinator with scontrol.
Dependency Matrix
Required Modules
None requiredComponents
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: multi-node-jax-ffi-debug Download link: https://github.com/KangOxford/auto-quant-research/archive/main.zip#multi-node-jax-ffi-debug Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.