Add util to get last time step of inference integration

jpc/__init__.py

@@ -11,6 +11,7 @@
 from ._utils import (
     get_fc_network as get_fc_network,
     compute_accuracy as compute_accuracy,
+    get_t_max as get_t_max,
     compute_pc_infer_energies as compute_pc_infer_energies
 )
 from ._train import (