4 Commits

Author	SHA1	Message	Date
Laurent Mazare	5320aa6b7d	Move the test-utils bits to a shared place. (#619)	2023-08-27 09:42:22 +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	0902846f25	Add the AdamW optimizer. (#307) ... * Add the AdamW optimizer. * Add some AdamW test validated against PyTorch.	2023-08-02 14:03:49 +01:00
Laurent Mazare	3eb2bc6d07	Softmax numerical stability. (#267) ... * Softmax numerical stability. * Fix the flash-attn test.	2023-07-28 13:13:01 +01:00