search

huggingface / candle

Minimalist ML framework for Rust
Apache License 2.0
13.79k stars 751 forks source link

Add argsort. #2132

Closed LaurentMazare closed 2 weeks ago

LaurentMazare commented 2 weeks ago

In this PR, this is done via a CustomOp1 but it's a bit unclear where this should live. I've put it in candle-core for now but maybe it belongs more in candle-nn?