Open mortont opened 7 months ago
Having Nx.argmax implemented will help to reduce backend_transfer calls when recursively calling Ortex for applications that use greedy search, beam search, etc.
Nx.argmax
backend_transfer
Having
Nx.argmaximplemented will help to reducebackend_transfercalls when recursively calling Ortex for applications that use greedy search, beam search, etc.