mx: support inference_mode and rank 3+ #3238
+39
−2
Merged
Loading