JAX Privacy
Getting Started
Overview
Installation
API Documentation
JAX & Flax API
Keras API
Examples
DP-SGD tutorial using Flax Linen on MNIST
Tutorial of DP-SGD LoRA fine-tuning Gemma3 in Keras on SAMSum dataset
Paper Results Reproduction
Paper Reproductions Guide
Technical Documentation
Troubleshooting
Library Design
Contribution Guide
Support
JAX Privacy
Index
Index
A
|
C
|
D
|
G
|
I
|
L
|
P
|
R
|
V
A
add_noise_to_grads() (jax_privacy.dp_sgd.gradients.DpsgdGradientComputer method)
(jax_privacy.dp_sgd.gradients.GradientComputer method)
C
calibrate_batch_size() (in module jax_privacy.accounting.calibrate)
calibrate_noise_multiplier() (in module jax_privacy.accounting.calibrate)
calibrate_num_updates() (in module jax_privacy.accounting.calibrate)
can_calibrate_batch_size() (jax_privacy.accounting.analysis.DpsgdTrainingAccountant method)
can_calibrate_noise_multipliers() (jax_privacy.accounting.analysis.DpsgdTrainingAccountant method)
can_calibrate_steps() (jax_privacy.accounting.analysis.DpsgdTrainingAccountant method)
clean_gradients() (jax_privacy.dp_sgd.gradients.GradientComputer method)
create_accountant() (jax_privacy.accounting.accountants.PldAccountantConfig method)
(jax_privacy.accounting.accountants.RdpAccountantConfig method)
D
DpsgdGradientComputer (class in jax_privacy.dp_sgd.gradients)
DpsgdTrainingAccountant (class in jax_privacy.accounting.analysis)
G
GradientComputer (class in jax_privacy.dp_sgd.gradients)
I
init_noise_state() (jax_privacy.dp_sgd.gradients.DpsgdGradientComputer method)
(jax_privacy.dp_sgd.gradients.GradientComputer method)
L
l2_loss() (jax_privacy.dp_sgd.gradients.GradientComputer method)
loss_and_clipped_gradients() (jax_privacy.dp_sgd.gradients.GradientComputer method)
P
PldAccountantConfig (class in jax_privacy.accounting.accountants)
R
RdpAccountantConfig (class in jax_privacy.accounting.accountants)
V
value_and_clipped_grad() (jax_privacy.dp_sgd.gradients.GradientComputer method)