diff --git a/python/mlx/nn/layers/__init__.py b/python/mlx/nn/layers/__init__.py index 26f77917fc..205a755fa6 100644 --- a/python/mlx/nn/layers/__init__.py +++ b/python/mlx/nn/layers/__init__.py @@ -18,6 +18,7 @@ from mlx.nn.layers.activations import ( ReLU6, Sigmoid, SiLU, + ClippedSiLU, Softmax, Softmin, Softplus, @@ -44,6 +45,7 @@ from mlx.nn.layers.activations import ( selu, sigmoid, silu, + clipped_silu, softmax, softmin, softplus,