Files
mlx-examples/flux/flux/autoencoder.py