jaxdpopt 0.0.1

Efficient DP optimization in JAX

Towards Efficient and Scalable Training of Differentially Private Deep Learning

(See '/research' for the code used in the benchmarks of our paper).

jaxdpopt package

Install using pip install . with Python>=3.10 in a fresh environment.

Examples can be found in '/examples' and tests can be run after installing pytest with python3 -m pytest ..

Contact

For comments and issues, create a new issue in the repository.

Citation

If you use this code, please cite our paper

@misc{beltran2024efficientscalabletrainingdifferentially,
      title={Towards Efficient and Scalable Training of Differentially Private Deep Learning}, 
      author={Sebastian Rodriguez Beltran and Marlon Tobaben and Joonas J{\"{a}}lk{\"{o}} and Niki Loppi and Antti Honkela},
      year={2024},
      eprint={2406.17298},
      archivePrefix={arXiv},
      primaryClass={cs.LG},
      url={https://arxiv.org/abs/2406.17298}, 
}