Closed horizon-blue closed 1 year ago
Summary: Previously, the mass matrices and some of the NUTS hyperparameters are not instantiated according to the dtype of the input, which can be problematic if the input tensors have double as dtype, or if the input tensor are on GPU.
Differential Revision: D40788741
This pull request was exported from Phabricator. Differential Revision: D40788741
Summary: Previously, the mass matrices and some of the NUTS hyperparameters are not instantiated according to the dtype of the input, which can be problematic if the input tensors have double as dtype, or if the input tensor are on GPU.
Differential Revision: D40788741