Currently using NumPyro for my Master’s thesis at the MPI for Biological Cybernetics under supervision of Peter Dayan.