jax
55512988 - tweak to fix xla_bridge.device_count behavior

Commit

6 years ago

tweak to fix xla_bridge.device_count behavior

References

#519 - add some pmap tests

Author

mattjj

mattjj

Parents

Loading