`n_trajectories` may need to be renamed to `batch_size`.

[hyeok9855]

Yes, n_trajectories is actually batch_size. I agree that this is confusing. But i think it's better to fix this in a separate PR.

Originally posted by @hyeok9855 in #431 (comment)

[josephdviviano]

yes - fair