Emile Aydar

icon company @ University College Dublin icon location Dublin icon location "Art is the only serious thing in the world. And the artist is the only person who is never serious" Oscar Wilde

Results 2 comments of Emile Aydar

Sparse MatMul Support in TorchJD for Seamless compatibility with torch.vmap()

> We already had a number of problems in the past due to `vmap`, so we decided to add the possibility to simply avoid it by setting `parallel_chunk_size=1` when calling...

Sparse MatMul Support in TorchJD for Seamless compatibility with torch.vmap()

> Also, do you think it could happen that people want to differentiate _with respect to_ a sparse tensor (e.g. have a sparse parameter tensor in their model)? In this...