This addresses a bug where a shape prevents a concatenate from creating the top-k from training set list of elements, and another one where an argmax is computed twice, making a gather operation fail when computing a top-k.
Divisions in this technique are also sometimes unstable so a small constant is added to stabilize them.
Docs
An explanation for a parameter in the bench module is added.
The docs are also updated to contain the links to the new notebooks.
Quick bug fix for RPS L2
Bug fix
This addresses a bug where a shape prevents a concatenate from creating the top-k from training set list of elements, and another one where an argmax is computed twice, making a gather operation fail when computing a top-k.
Divisions in this technique are also sometimes unstable so a small constant is added to stabilize them.
Docs
An explanation for a parameter in the bench module is added.
The docs are also updated to contain the links to the new notebooks.