Remix.run Logo
PyTorch Landscape(pytorch.landscape2.io)
29 points by salamo 4 hours ago | 4 comments
fouronnes3 an hour ago | parent | next [-]

I work on differentiable geometric optics with PyTorch. Seeing a list like this is really illustrative of the power that PyTorch provides when you start considering it like a general purpose GPU-enabled state of the art numerical optimization framework.

One thing I wonder is why no one has made a fork of PyTorch yet that removes all the API surface that doesn't produce GPU friendly code. Make dtype and device arg mandatory without defaults, remove in place operations that trigger a CPU sync, etc. This would increase confidence that written code will run on the GPU and pass torch.export() on the first try.

rhdunn 24 minutes ago | parent [-]

It's useful to be able to run models/code on the CPU or split between CPU and GPU, especially for models that cannot fit into the GPU's VRAM, if you are running multiple models, or if you have training data that you need to move between CPU and GPU.

d3m0t3p an hour ago | parent | prev | next [-]

Interesting to see clearml but not its bigger counterpart mlflow

yurimo an hour ago | parent | prev [-]

Great stuff, thanks for posting.