Random

Debugging is not fun (1/n)

If you think you know JAX and are confident in your debugging skills, try this challenge:

JAX Debugging Colab Notebook

Just a 100 line notebook that implements a simple ML algorithm (in this case, flow matching), but does something very weird. What could possibly be wrong? I once had to spend a whole weekend debugging because of the same root issue as the one in this notebook.