Train/Infer Sparse Models on TX2?

Hi,

Please check this comment to add the TX2 GPU architecture and recompile the library.

Thanks.