Small improvement on dtype specification

This corresponds to the merge of #43 by @chaithyagr

batched-gradients-trajectory

This releases corresponds to correcting the gradients of the trajectory for the batch case by @chaithyagr

See PR #40

It is also the first release that will automatically trigger a build and a push to pypi (hopefully)