JAX Privacy
Getting Started
Overview
Installation
API Documentation
Core Library
Individual Components
jax_privacy.accounting
jax_privacy.batch_selection
jax_privacy.clipping
jax_privacy.clipping.clip_and_round_to_grid
jax_privacy.clipping.clip_pytree
jax_privacy.clipping.clipped_fun
jax_privacy.clipping.clipped_grad
jax_privacy.clipping.AuxiliaryOutput
jax_privacy.clipping.BoundedSensitivityCallable
jax_privacy.noise_addition
jax_privacy.optimizers
jax_privacy.auditing
jax_privacy.matrix_factorization
Higher-level APIs
Experimental Modules
Examples
Keras API
Examples Guide
DP-SGD tutorial using Flax Linen on MNIST
Tutorial of DP-SGD LoRA fine-tuning Gemma3 in Keras on SAMSum dataset
Tutorial: Generating Differentially Private Synthetic Data
Sharp Edges
Handling Variable Batch Sizes
VMap + Sharding
Mixed Precision Training
Technical Documentation
Paper Reproductions Guide
Troubleshooting
Library Design
Contribution Guide
Support
JAX Privacy
Core Library
jax_privacy.clipping
jax_privacy.clipping.AuxiliaryOutput
View page source
jax_privacy.clipping.AuxiliaryOutput
jax_privacy.clipping.
AuxiliaryOutput
alias of
Aux