summaryrefslogtreecommitdiff
path: root/candle-nn/tests/ops.rs
Commit message (Expand)AuthorAgeFilesLines
* Improved launch config for layer-norm/rms-norm. (#2591)Laurent Mazare2024-11-041-0/+45
* Add the layernorm specialized op. (#2212)Laurent Mazare2024-05-241-0/+27
* Fix sigmoid gradient calculation and move sigmoid into a specialized op (#2114)MilkFather2024-04-291-0/+11
* Add the rope THD kernel. (#2014)Laurent Mazare2024-04-051-0/+31
* Contiguous variant of the rope kernel. (#1929)Laurent Mazare2024-03-251-2/+30
* Fast kernels for rotary embeddings. (#1928)Laurent Mazare2024-03-241-0/+28
* Custom op for RmsNorm (#1890)Laurent Mazare2024-03-211-4/+30
* Add a custom softmax implementation. (#744)Laurent Mazare2023-09-051-0/+10
* Move the test-utils bits to a shared place. (#619)Laurent Mazare2023-08-271-7/+4
* Add a yolo-v3 example. (#528)Laurent Mazare2023-08-201-0/+3
* Add the AdamW optimizer. (#307)Laurent Mazare2023-08-021-14/+6
* Softmax numerical stability. (#267)Laurent Mazare2023-07-281-0/+62