.gitignore
LICENSE
README.md
pyproject.toml
setup.py
examples/example.ipynb
examples/example_distributed.ipynb
examples/example_fixed_bs.ipynb
jaxdpopt/data.py
jaxdpopt/dp_accounting_utils.py
jaxdpopt/jax_mask_efficient.py
jaxdpopt/models.py
jaxdpopt.egg-info/PKG-INFO
jaxdpopt.egg-info/SOURCES.txt
jaxdpopt.egg-info/dependency_links.txt
jaxdpopt.egg-info/requires.txt
jaxdpopt.egg-info/top_level.txt
research/README.MD
research/requirements.txt
research/jax_exp/GenericBatchManager.py
research/jax_exp/data.py
research/jax_exp/jax_mask_efficient.py
research/jax_exp/models.py
research/jax_exp/naive_jax_pipeline.py
research/jax_exp/thr_record.py
research/jax_exp/thr_record_for_mask.py
research/torch_exp/GenericBatchManager.py
research/torch_exp/data.py
research/torch_exp/model_functions.py
research/torch_exp/models.py
research/torch_exp/pipeline_torch.py
research/torch_exp/privacy_engines.py
research/torch_exp/seeding_utils.py
research/torch_exp/thr_record.py
tests/test_data.py
tests/test_dp_accounting.py
tests/test_end_to_end_simple.py
tests/test_jax_mask_efficient.py
tests/test_models.py