Xiu-zhe (Roger) Luo
Xiu-zhe (Roger) Luo
@MikeInnes I personally prefer `cuda` instead of `gpu` tho, pytorch use this as well. Regarding to treelike, we already have one (tho, it's a bit ugly at the moment), it...
@MikeInnes What about AbstractTree.jl, I copied part of it while implementing our printings. Is there any reason Flux is not using this? Maybe we should just have some functions instead...
@MikeInnes I see. I guess you only need that for the conversion between CPU and GPU in Flux. We do need some traverse API to dispatch/collect parameters however, unlike normal...
@MikeInnes I'd love to have `cuda` for converting to `CuArray` and `opencl`/`cl` for converting to `CLArray` and maybe also `tpu` for converting to `XRTArray`. And they are also simple bindings...
Thanks for the note! we recently have an efficient CUDA implementation of time evolution using Krylov to be open sourcing soon, we will integrate that implementation to solve this issue...
FYI `DitStr` is technically a "ieee-like integer" so it overflows instead of an error.
@JuliaRegistrator register()
@JuliaRegistrator register
@JuliaRegistrator register
@JuliaRegistrator register