Loading...
Loading...
Loading...
5 rules available in the Claude directory
Unlock JAX best practices to supercharge your Python code for ML workloads. Learn functional programming, JIT compilation, vmap vectorization, and pure functions for optimal speed and compatibility.
Advanced prompt for custom physics simulations, PDE solvers, and differentiable scientific computing in JAX.
Specialized prompt for architecting scalable ML models with JAX, Flax, and Optax, optimized for research workflows.
Comprehensive system prompt for developing high-performance, differentiable numerical programs in JAX using Claude's long context and reasoning.
Adapted guidelines for writing high-performance JAX code in Python, optimized for Claude Code CLI's long context and tool integration.