Rishi Puri

icon indicating a website link https://www.linkedin.com/in/rishi-puri-1726b1147/ icon indicating an email [email protected]

icon company NVIDIA icon location SF Bay Area icon location Engineering Lead for PyTorch Geometric (PyG) @ NVIDIA. I am a Core Contributor for PyG

Results 44 comments of Rishi Puri

`OGBMAG240m` example

@rusty1s @akihironitta lmk if there is anything else i would need to get this merge ready

`OGBMAG240m` example

@akihironitta thanks for the review, no rush, please merge once your ready.

[Dataset] add MedShapeNet to datasets

I will guide @jdhenaos and @loupl in completing this task

[Dataset] add MedShapeNet to datasets

@ericauld as well

GPU memory linear growing up without `torch.cuda.empty_cache()`

@shaochengyan @rusty1s seems this issue can be closed?

Add G-retriever (GNN+LLM) example

@akihironitta anything else needed to merge?

Add G-retriever (GNN+LLM) example

@rusty1s @akihironitta happy to make any required changes to get this merged finally

Add pcqm4m multi-gpu example: large scale graph regression

@akihironitta let us know if anything else needed to get this merged?

segment_matmul failing on CUDA

I tested our latest NVIDIA container on ``` nvidia-smi -L GPU 0: NVIDIA GeForce GTX 1080 Ti (UUID: GPU-89a48638-3d6b-10ae-38b1-01f03b50d9e8) ``` and cannot reproduce the error. i will see if older...

segment_matmul failing on CUDA

when i tried to build older versions i hit some build issues. @dfdazac is it possible for you to try our containers and see if they work for you? this...