https://github.com/google/jax
History
Tip revision: 500da57e91eb4b9094eb2b5cfa0826691862417f authored by jax authors on 07 May 2024, 14:07:04 UTC
Merge pull request #21077 from merrymercy:patch-1
Tip revision: 500da57
File Mode Size
gpu_ops.cpp -rw-r--r-- 1.6 KB
kernel_helpers.h -rw-r--r-- 2.2 KB
kernels.h -rw-r--r-- 1.3 KB
pybind11_kernel_helpers.h -rw-r--r-- 1.5 KB
rms_norm_kernels.cu -rw-r--r-- 39.0 KB

back to top