peng-lab/BaSiCPy

JAX to accelerate computations

Closed this issue · 4 comments

https://github.com/google/jax

includes SVD and DCT
Numpy like interface
JIT compiling
I think the same code should work for cpu/gpu/tpu

I think this is an excellent idea.

See also #17

closed by #62