* update to latest mlx swift and use fast norms * gpu usage -> memory usage
* LeNet on MNIST + readme update * tanh + remove device toggle * remove device entirely