Write correct, idiomatic Apple MLX code for Apple Silicon ML. Use when working with MLX arrays, neural networks, training loops, lazy evaluation, unified memory, mx.eval, mx.compile, Metal GPU, memory
development/mlx-dev
Use when you have lint errors, formatting issues, or before committing code to ensure it passes CI.
This skill should be used when the user asks to "update documentation for my changes", "check docs for this PR", "what d...
Write docstrings for PyTorch functions and methods following PyTorch conventions. Use when writing or updating docstring...