Hugging Face
Models
Datasets
Spaces
Posts
Docs
Enterprise
Pricing
Log In
Sign Up
akswelh
/
NEOX
like
0
arxiv:
29 papers
Model card
Files
Files and versions
Community
main
NEOX
/
megatron
/
fused_kernels
1 contributor
History:
1 commit
akswelh
Upload 251 files
d90b3a8
verified
2 months ago
__init__.py
6.03 kB
Upload 251 files
2 months ago
compat.h
893 Bytes
Upload 251 files
2 months ago
fused_rotary_positional_embedding.cpp
6.53 kB
Upload 251 files
2 months ago
fused_rotary_positional_embedding.h
19.1 kB
Upload 251 files
2 months ago
fused_rotary_positional_embedding_cuda.cu
15.7 kB
Upload 251 files
2 months ago
scaled_masked_softmax.cpp
3.21 kB
Upload 251 files
2 months ago
scaled_masked_softmax.h
24 kB
Upload 251 files
2 months ago
scaled_masked_softmax_cuda.cu
4.66 kB
Upload 251 files
2 months ago
scaled_upper_triang_masked_softmax.cpp
2.7 kB
Upload 251 files
2 months ago
scaled_upper_triang_masked_softmax.h
26.9 kB
Upload 251 files
2 months ago
scaled_upper_triang_masked_softmax_cuda.cu
3.45 kB
Upload 251 files
2 months ago
setup.py
2.99 kB
Upload 251 files
2 months ago
type_shim.h
22.1 kB
Upload 251 files
2 months ago
