Awni Hannun c3628eea49 Add mx.finfo and use it when making causal mask (#1726) ...

* finfo

* fixes

* docs

2024-12-19 14:52:41 -08:00

..

__init__.py

Adds 3D pooling (#1526)

2024-11-19 16:45:24 -08:00

activations.py

feat: Added "tanh" option to GELU approximation (#1268)

2024-07-28 09:07:56 +02:00

base.py

fix module attribute set, reset, set (#1403)

2024-09-11 16:30:42 -07:00

containers.py

copyright + ack

2023-11-30 11:12:53 -08:00

convolution_transpose.py

Transposed Convolution (#1245)

2024-09-06 19:52:38 -07:00

convolution.py

add groups in conv2d (#1569)

2024-11-07 13:57:53 -08:00

dropout.py

Typing the dropout. (#1479)

2024-10-15 06:45:46 -07:00

embedding.py

Block sparse qmm (#1124)

2024-05-16 15:24:14 -07:00

linear.py

Block sparse qmm (#1124)

2024-05-16 15:24:14 -07:00

normalization.py

faster group norm (#1304)

2024-08-01 12:49:23 -07:00

pooling.py

Doc fix (#1615)

2024-11-22 11:12:25 -08:00

positional_encoding.py

Doc error for default for scale in SinusoidalPositionalEncoding (#1174)

2024-06-02 13:42:45 -07:00

quantized.py

No reshapes in quantized embedding (#1682)

2024-12-09 18:57:38 -08:00

recurrent.py

Faster RNN layers (#1419)

2024-09-17 06:04:19 -07:00

transformer.py

Add mx.finfo and use it when making causal mask (#1726)

2024-12-19 14:52:41 -08:00

upsample.py

add cubic to type hinting for upsample (#1709)

2024-12-17 07:30:23 -08:00