4 Commits

Author	SHA1	Message	Date
Laurent Mazare	7529531056	Add the optimizer trait. (#702)	2023-09-01 12:55:39 +01:00
Laurent Mazare	a1812f934f	Add a yolo-v3 example. (#528) ... * 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.	2023-08-20 18:19:37 +01:00
Laurent Mazare	c78ce76501	Add a simple Module trait and implement it for the various nn layers (#500) ... * Start adding the module trait. * Use the module trait. * Implement module for qmatmul.	2023-08-18 09:38:22 +01:00
Laurent Mazare	8ad4a21ffc	Add a basic optimizer example. (#454)	2023-08-15 17:19:18 +01:00