Commit message (Collapse) | Author | Age | Files | Lines | |
---|---|---|---|---|---|
* | Add the optimizer trait. (#702) | Laurent Mazare | 2023-09-01 | 1 | -1/+1 |
| | |||||
* | Add a yolo-v3 example. (#528) | Laurent Mazare | 2023-08-20 | 1 | -0/+6 |
| | | | | | | | | | | | | | | | * Add a couple functions required for yolo. * Add the yolo-v3 example. * Add minimum and maximum. * Use the newly introduced maximum. * Cuda support for min/max + add some testing. * Allow for more tests to work with accelerate. * Fix a typo. | ||||
* | Add a simple Module trait and implement it for the various nn layers (#500) | Laurent Mazare | 2023-08-18 | 1 | -1/+1 |
| | | | | | | | * Start adding the module trait. * Use the module trait. * Implement module for qmatmul. | ||||
* | Add a basic optimizer example. (#454) | Laurent Mazare | 2023-08-15 | 1 | -0/+33 |