akswelh
/

NEOX

Model card Files Files and versions Community

NEOX / megatron /model /mamba /__init__.py

akswelh's picture

Upload 251 files

d90b3a8 verified about 1 month ago

91 Bytes

	from .mamba import (
	ParallelMambaResidualLayer,
	ParallelMambaResidualLayerPipe,
	)