3 Commits

Author SHA1 Message Date
Peter Hawkins d988ff1bf2 Replaces references to jax.numpy.DeviceArray with jax.Array.
PiperOrigin-RevId: 515678285
2023-06-02 18:03:13 +01:00
Peter Hawkins 784f67565e Suppress some pytype errors related to jnp.DeviceArray == jax.Array.
PiperOrigin-RevId: 514986169
2023-06-02 18:02:45 +01:00
Ira Ktena cff83be778 Adding counterfactual_fairness to deepmind-research repo
PiperOrigin-RevId: 396665335
2021-09-17 17:53:02 +01:00