updateΒΆ

Auto-generated from flax_nnx_code_defs

PyTorch

API: torch.nn.modules.container.ParameterDict.update
Strategy: Direct Mapping

Apple MLX

API: mlx.optimizers.optimizers.Optimizer.update
Strategy: Direct Mapping

Flax NNX

API: nnx.training.optimizer.ModelAndOptimizer.update
Strategy: Direct Mapping