import Playgrounds
import MLX

#Playground("Testing MLXArray") {
   let matrix = MLXArray([9, 2, 3, 0, 5, 6], [2, 3])
   // array([[9, 2, 3],
   //        [0, 5, 6]], dtype=int32)

   let all = all(matrix)
   // array(false, dtype=bool)
   
   let max = max(matrix, axis: 0)
   // array([9, 5, 6], dtype=int32)
}