Chime Ogbuji 3b661b7394 Add linear warmup and schedule joining for use with existing schedules (#721) ... * Add linear warmup to schedules for use with existing schedules * Changed parameters for simplicity of most common case (0 initial value) * Added ScheduleJoiner and updated documentation * ScheduleJoiner -> join_schedules (ala optax #) * black compliance * Different evaluation of schedules * nits --------- Co-authored-by: Awni Hannun <awni@apple.com>		2024-02-26 07:28:48 -08:00
..
mlx	Add linear warmup and schedule joining for use with existing schedules (#721)	2024-02-26 07:28:48 -08:00
src	Shapeless compilation for some graphs (#687)	2024-02-19 21:43:54 -08:00
tests	Add linear warmup and schedule joining for use with existing schedules (#721)	2024-02-26 07:28:48 -08:00