parametersΒΆ

Auto-generated from flax_nnx_code_defs

PyTorch

API: torch.nn.Module.parameters
Strategy: Direct Mapping

Keras

API: keras.layers.TorchModuleWrapper.parameters
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.layers.base.Module.parameters
Strategy: Direct Mapping

Flax NNX

API: torch_parameters_to_nnx
Strategy: Plugin (torch_parameters_to_nnx)