mirror of
https://github.com/ml-explore/mlx.git
synced 2025-08-07 23:41:35 +08:00
adding to __init__.py
This commit is contained in:
parent
a426880baf
commit
9cb6df5960
@ -18,6 +18,7 @@ from mlx.nn.layers.activations import (
|
|||||||
ReLU6,
|
ReLU6,
|
||||||
Sigmoid,
|
Sigmoid,
|
||||||
SiLU,
|
SiLU,
|
||||||
|
ClippedSiLU,
|
||||||
Softmax,
|
Softmax,
|
||||||
Softmin,
|
Softmin,
|
||||||
Softplus,
|
Softplus,
|
||||||
@ -44,6 +45,7 @@ from mlx.nn.layers.activations import (
|
|||||||
selu,
|
selu,
|
||||||
sigmoid,
|
sigmoid,
|
||||||
silu,
|
silu,
|
||||||
|
clipped_silu,
|
||||||
softmax,
|
softmax,
|
||||||
softmin,
|
softmin,
|
||||||
softplus,
|
softplus,
|
||||||
|
Loading…
Reference in New Issue
Block a user