Chen Ao

icon location Fudan University BSc Physics -> ETH Zurich MSc Physics -> University of Augsburg PhD Physics (in progress). Studying neural network quantum states.

Results 1 comments of Chen Ao

`jax.lax.dot` with `preferred_element_type` gives wrong result

I did some further tests in these days. This bug doesn't happen on CPU, but still happens on A100 GPU.