nonlinearityΒΆ

Auto-generated from mlx_code_defs

PyTorch

API: torch.nn.modules.rnn.RNNCell.nonlinearity
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.layers.recurrent.RNN.nonlinearity
Strategy: Direct Mapping