jax
Adding gpu_device_targets parameter
#233
Merged

Loading