cythonize featurizer.
proteneer opened this issue · comments
Yutong Zhao commented
Currently the featurizer, even though it's written as pure TF ops code, is still painfully slow. We can probably get a pretty big win by writing some simple cython extensions.
Yutong Zhao commented
ported to pure gpu code