PyTorch and MLX for Apple Silicon

Comparing and implementing a convolutional neural network using PyTorch and MLX for Apple Silicon. The MLX version of the code is almost identical to the PyTorch version with a few differences. MLX has interesting properties like the use of the 'Array' class, lazy computation, and compilation for optimizing function execution.