By rewriting the Trevisan extractor in JAX, we can run the extractor on the GPU.
By rewriting the Trevisan extractor in JAX, we can run the extractor on the GPU.