JAX Privacy
Getting Started
Overview
Installation
API Documentation
Core Library
Individual Components
jax_privacy.accounting
jax_privacy.batch_selection
jax_privacy.clipping
jax_privacy.noise_addition
jax_privacy.optimizers
jax_privacy.auditing
jax_privacy.matrix_factorization
jax_privacy.matrix_factorization.banded
jax_privacy.matrix_factorization.buffered_toeplitz
jax_privacy.matrix_factorization.dense
jax_privacy.matrix_factorization.optimization
jax_privacy.matrix_factorization.sensitivity
jax_privacy.matrix_factorization.streaming_matrix
jax_privacy.matrix_factorization.toeplitz
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.matrix_factorization
jax_privacy.matrix_factorization.optimization
jax_privacy.matrix_factorization.optimization.optimize
View page source
jax_privacy.matrix_factorization.optimization.optimize
jax_privacy.matrix_factorization.optimization.
optimize
(
*
args
,
**
kwargs
)
[source]