7 Commits

Author	SHA1	Message	Date
Awni Hannun	3dcb286baf	Remove stream from average grads so it uses default (#2532) ... * Remove stream from average grads so it uses default * comment	2025-08-25 15:56:29 -07:00
Anastasiia Filippova	9392fc3f88	NCCL backend (#2476)	2025-08-21 11:56:15 -07:00
Angelos Katharopoulos	914409fef9	Data parallel helper (#1407)	2024-09-16 18:17:21 -07:00
Awni Hannun	291cf40aca	Some fixes to typing (#1371) ... * some fixes to typing * fix module reference * comment	2024-08-28 11:16:19 -07:00
Angelos Katharopoulos	0de5988f92	Custom VJP and checkpointing (#541) ... * Implement custom_vjp and checkpointing * Add a dependency management primitive * Change the eval order to deep branches first * Add graph depth tracking to the array	2024-01-30 16:04:45 -08:00
Awni Hannun	46a39e5b1f	copyright + ack	2023-11-30 11:12:53 -08:00
Awni Hannun	8ca7f9e8e9	awni's commit files	2023-11-29 10:30:41 -08:00