index Distribution using jnp.indices instead of np.indices#292
Open
adzcai wants to merge 1 commit intogoogle-deepmind:mainfrom
Open
index Distribution using jnp.indices instead of np.indices#292adzcai wants to merge 1 commit intogoogle-deepmind:mainfrom
adzcai wants to merge 1 commit intogoogle-deepmind:mainfrom
