mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 01:41:17 +08:00
Sort some APIs docs by names (a-z) (#472)
This commit is contained in:
parent
a2bf7693dd
commit
4a9012cba0
@ -15,9 +15,9 @@ simple functions.
|
|||||||
gelu
|
gelu
|
||||||
gelu_approx
|
gelu_approx
|
||||||
gelu_fast_approx
|
gelu_fast_approx
|
||||||
relu
|
mish
|
||||||
prelu
|
prelu
|
||||||
|
relu
|
||||||
|
selu
|
||||||
silu
|
silu
|
||||||
step
|
step
|
||||||
selu
|
|
||||||
mish
|
|
||||||
|
@ -9,29 +9,29 @@ Layers
|
|||||||
:toctree: _autosummary
|
:toctree: _autosummary
|
||||||
:template: nn-module-template.rst
|
:template: nn-module-template.rst
|
||||||
|
|
||||||
Sequential
|
ALiBi
|
||||||
ReLU
|
BatchNorm
|
||||||
PReLU
|
|
||||||
GELU
|
|
||||||
SiLU
|
|
||||||
Step
|
|
||||||
SELU
|
|
||||||
Mish
|
|
||||||
Embedding
|
|
||||||
Linear
|
|
||||||
QuantizedLinear
|
|
||||||
Conv1d
|
Conv1d
|
||||||
Conv2d
|
Conv2d
|
||||||
BatchNorm
|
|
||||||
LayerNorm
|
|
||||||
RMSNorm
|
|
||||||
GroupNorm
|
|
||||||
InstanceNorm
|
|
||||||
Dropout
|
Dropout
|
||||||
Dropout2d
|
Dropout2d
|
||||||
Dropout3d
|
Dropout3d
|
||||||
Transformer
|
Embedding
|
||||||
|
GELU
|
||||||
|
GroupNorm
|
||||||
|
InstanceNorm
|
||||||
|
LayerNorm
|
||||||
|
Linear
|
||||||
|
Mish
|
||||||
MultiHeadAttention
|
MultiHeadAttention
|
||||||
ALiBi
|
PReLU
|
||||||
|
QuantizedLinear
|
||||||
|
RMSNorm
|
||||||
|
ReLU
|
||||||
RoPE
|
RoPE
|
||||||
|
SELU
|
||||||
|
Sequential
|
||||||
|
SiLU
|
||||||
SinusoidalPositionalEncoding
|
SinusoidalPositionalEncoding
|
||||||
|
Step
|
||||||
|
Transformer
|
||||||
|
@ -10,14 +10,14 @@ Loss Functions
|
|||||||
:template: nn-module-template.rst
|
:template: nn-module-template.rst
|
||||||
|
|
||||||
binary_cross_entropy
|
binary_cross_entropy
|
||||||
|
cosine_similarity_loss
|
||||||
cross_entropy
|
cross_entropy
|
||||||
|
hinge_loss
|
||||||
|
huber_loss
|
||||||
kl_div_loss
|
kl_div_loss
|
||||||
l1_loss
|
l1_loss
|
||||||
|
log_cosh_loss
|
||||||
mse_loss
|
mse_loss
|
||||||
nll_loss
|
nll_loss
|
||||||
smooth_l1_loss
|
smooth_l1_loss
|
||||||
triplet_loss
|
triplet_loss
|
||||||
hinge_loss
|
|
||||||
huber_loss
|
|
||||||
log_cosh_loss
|
|
||||||
cosine_similarity_loss
|
|
@ -33,13 +33,13 @@ we use a splittable version of Threefry, which is a counter-based PRNG.
|
|||||||
.. autosummary::
|
.. autosummary::
|
||||||
:toctree: _autosummary
|
:toctree: _autosummary
|
||||||
|
|
||||||
seed
|
|
||||||
key
|
|
||||||
split
|
|
||||||
bernoulli
|
bernoulli
|
||||||
categorical
|
categorical
|
||||||
gumbel
|
gumbel
|
||||||
|
key
|
||||||
normal
|
normal
|
||||||
randint
|
randint
|
||||||
uniform
|
seed
|
||||||
|
split
|
||||||
truncated_normal
|
truncated_normal
|
||||||
|
uniform
|
||||||
|
Loading…
Reference in New Issue
Block a user