Name		Name	Last commit message	Last commit date
.gitignore		.gitignore
CMakeLists.txt		CMakeLists.txt
README.md		README.md
bench.cpp		bench.cpp
bench.py		bench.py
kernel_helpers.h		kernel_helpers.h
kernel_pybind11_helpers.h		kernel_pybind11_helpers.h
makefile		makefile
threefry.py		threefry.py
threefry_avx.cpp		threefry_avx.cpp
threefry_avx.h		threefry_avx.h
tsc_x86.h		tsc_x86.h

Repository files navigation

AVX threefry kernels for jax (WIP)

iirc roughly 1.7 times faster than the autovectorised ones in in the tf part of jaxlib (which already works amazingly well)
achieved by upping ILP through manual unrolling

About

No description, website, or topics provided.

0 stars

1 watching

Report repository

Releases

Packages

Contributors

Languages