the z-scoring tests need some refactoring:

• variable names are confusing, e.g., net should be called build_fun
• there are many duplicate test cases
• why pass net(t_batch, t_batch) and not something like net(theta, x).
• add a case for mnle