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.argmax
implemented will help to reducebackend_transfer
calls when recursively calling Ortex for applications that use greedy search, beam search, etc.