[0, 10] → Gini = 0.5 for n=2.

All-equal values → Gini = 0.

Unknown model_type should not cause MoE classification.

Unloading when nothing is loaded should not raise.

Variance explained should be between 0 and 1.

Wasserstein-optimal should have lower cost ratio than diff-in-means.

Whitened SVD should produce orthogonal directions.

Harmful and harmless with different hidden dims should fail or handle gracefully.

WhitenedSVD with NaN — currently raises; documenting behavior.

Single sample should not crash (may return 0 directions due to insufficient data).

WhitenedSVD with all-zero activations.

Test with provided attention and MLP outputs.

Should wrap prompts when tokenizer has apply_chat_template.

Zero matrix should have effective rank 0.

A legitimate score of 0.0 should be returned, not treated as falsy.

A clean temporary output directory for test artifacts.

Convert a PyTorch tensor to an MLX array.