lightning-thunder icon indicating copy to clipboard operation
lightning-thunder copied to clipboard

Published 20 hours ago • verified publisher icon Lightning-AI

Implement _VariableFunctionsClass.max of torch

Open athitten opened this issue 1 year ago • 1 comments

🚀 Feature

Implement torch.max

Motivation

NeMo's MegatronNeVaModel

cc @tfogal

athitten avatar May 01 '24 18:05 athitten

@athitten Do you know how max is called in NeVa?

mruberry avatar May 06 '24 16:05 mruberry