Files
mlx-examples/normalizing_flow/flows.py