[CUDA] Fsdp (easy) (#3130) · 72e04f7fb7 - mlx - L'Électron Rare — Forge

L-electron-Rare/mlx

[CUDA] Fsdp (easy) (#3130)

This commit is contained in:

Anastasiia Filippova

2026-03-01 23:29:09 +01:00

committed by

GitHub

parent 6482d13dd3

commit 72e04f7fb7

4 changed files with 311 additions and 20 deletions

									
										docs/src/python/nn.rst
									
		+1
		
												View File
												
				@@ -175,6 +175,7 @@ In detail:

				   value_and_grad

				   quantize

				   average_gradients

				   fsdp_apply_gradients

				.. toctree::