Index of /img/APP/home/jetson/python/lib/python3.10/site-packages/transformer_engine/jax/
Name ↑
Size
Modified
⬆
..
📁
__pycache__
-
2025-06-02 10:59
📁
cpp_extensions
-
2025-06-02 10:59
📁
flax
-
2025-06-02 10:59
📁
praxis
-
2025-06-02 10:59
📁
quantize
-
2025-06-02 10:59
📜
__init__.py
4.0 KB
2025-06-02 10:59
📜
activation.py
2.8 KB
2025-06-02 10:59
📜
attention.py
42.9 KB
2025-06-02 10:59
📜
dense.py
10.7 KB
2025-06-02 10:59
📜
layernorm.py
4.3 KB
2025-06-02 10:59
📜
layernorm_dense.py
9.1 KB
2025-06-02 10:59
📜
layernorm_mlp.py
15.6 KB
2025-06-02 10:59
📜
setup.py
3.4 KB
2025-06-02 10:59
📜
sharding.py
10.3 KB
2025-06-02 10:59
📜
softmax.py
2.0 KB
2025-06-02 10:59