CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
ai-forever

Real-time collaboration for Jupyter Notebooks, Linux Terminals, LaTeX, VS Code, R IDE, and more,
all in one place. Commercial Alternative to JupyterHub.

GitHub Repository: ai-forever/sber-swap
 Path: tree/main/apex/csrc
Views: 792
Name
Size
Last Modified
amp_C_frontend.cpp4.1 KB
compat.h140 bytes
flatten_unflatten.cpp584 bytes
layer_norm_cuda.cpp6.5 KB
layer_norm_cuda_kernel.cu25 KB
mlp.cpp4.6 KB
mlp_cuda.cu52.3 KB
multi_tensor_adagrad.cu3 KB
multi_tensor_adam.cu4.5 KB
multi_tensor_apply.cuh4.7 KB
multi_tensor_axpby_kernel.cu4.7 KB
multi_tensor_l2norm_kernel.cu12.8 KB
multi_tensor_lamb.cu12.6 KB
multi_tensor_lamb_stage_1.cu4.4 KB
multi_tensor_lamb_stage_2.cu3.4 KB
multi_tensor_novograd.cu5.1 KB
multi_tensor_scale_kernel.cu4.1 KB
multi_tensor_sgd_kernel.cu8.1 KB
syncbn.cpp6.2 KB
type_shim.h4.6 KB
welford.cu54.6 KB