bench: refactor keras_api.py to use Poisson Sampling #89

Neerajpathak07 wants to merge 5 commits intogoogle-deepmind:maingoogle-deepmind/jax_privacy:mainfrom

Neerajpathak07:poisson-samplingNeerajpathak07/jax_privacy:poisson-samplingCopy head branch name to clipboard