4 Commits

Author	SHA1	Message	Date
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	eab54e4490	Fix the tests for mkl. (#437)	2023-08-14 08:09:27 +01:00
Laurent Mazare	1064b9b031	Add the cross-entropy loss. (#287)	2023-07-31 14:26:36 +01:00
Laurent Mazare	ffeafbfc43	Make the nll op closer to the pytorch version + add a test. (#286)	2023-07-31 14:14:01 +01:00