Hugging Face
Models
Datasets
Spaces
Buckets
new
Docs
Enterprise
Pricing
Website
Tasks
HuggingChat
Collections
Languages
Organizations
Community
Blog
Posts
Daily Papers
Learn
Discord
Forum
GitHub
Solutions
Team & Enterprise
Hugging Face PRO
Enterprise Support
Inference Providers
Inference Endpoints
Storage Buckets
Log In
Sign Up
camenduru
/
apex
like
1
Model card
Files
Files and versions
xet
Community
Copy to bucket
new
main
apex
/
csrc
/
megatron
105 kB
Ctrl+K
Ctrl+K
1 contributor
History:
1 commit
camenduru
thanks to NVIDIA ❤
e828767
about 3 years ago
fused_weight_gradient_dense.cpp
540 Bytes
thanks to NVIDIA ❤
about 3 years ago
fused_weight_gradient_dense_16bit_prec_cuda.cu
3.52 kB
thanks to NVIDIA ❤
about 3 years ago
fused_weight_gradient_dense_cuda.cu
4.28 kB
thanks to NVIDIA ❤
about 3 years ago
generic_scaled_masked_softmax.cpp
2.95 kB
thanks to NVIDIA ❤
about 3 years ago
generic_scaled_masked_softmax.h
12.6 kB
thanks to NVIDIA ❤
about 3 years ago
generic_scaled_masked_softmax_cuda.cu
3.83 kB
thanks to NVIDIA ❤
about 3 years ago
scaled_masked_softmax.cpp
3.37 kB
thanks to NVIDIA ❤
about 3 years ago
scaled_masked_softmax.h
32.7 kB
thanks to NVIDIA ❤
about 3 years ago
scaled_masked_softmax_cuda.cu
4.21 kB
thanks to NVIDIA ❤
about 3 years ago
scaled_softmax.cpp
2.47 kB
thanks to NVIDIA ❤
about 3 years ago
scaled_softmax_cuda.cu
3.28 kB
thanks to NVIDIA ❤
about 3 years ago
scaled_upper_triang_masked_softmax.cpp
2.55 kB
thanks to NVIDIA ❤
about 3 years ago
scaled_upper_triang_masked_softmax.h
25.3 kB
thanks to NVIDIA ❤
about 3 years ago
scaled_upper_triang_masked_softmax_cuda.cu
3.16 kB
thanks to NVIDIA ❤
about 3 years ago
