jax
porting changes from 0.8.0 to 0.8.2
#601
Merged

Loading