Comment by tucan9389

Thank you. Agreed, we should be more careful when doing this. Was it the JAX training codebase?