I think you’re reading way too much into a simple minim-finding script: https://github.com/kingoflolz/mesh-transformer-jax/blob/master/device_train.py
I think you’re reading way too much into a simple minim-finding script: https://github.com/kingoflolz/mesh-transformer-jax/blob/master/device_train.py