akswelh
/

NEOX

Model card Files Files and versions Community

NEOX / requirements /requirements-mamba.txt

akswelh's picture

Upload 251 files

d90b3a8 verified 24 days ago

104 Bytes

	causal_conv1d>=1.1.0
	einops
	mamba_ssm>=1.2.0.post1 # required for untied embedding + unembedding layers