Hi Chris!
Can you try without njobs=2? I think that will cause problems on a single GPU. Also, make sure to update to recent master as @aseyboldt did some refactoring of NUTS to help with GPU acceleration.
Best,
Thomas
1 Like