mrq c494894261 simple DDP wrapper (for my NVlink test)		2024-05-04 11:48:26 -05:00
..
__init__.py
adaln.py
ar_nar.py	simple DDP wrapper (for my NVlink test)	2024-05-04 11:48:26 -05:00
base.py
retnet_hf.py
retnet_ts.py
retnet.py
transformer.py