JAX Privacy

Getting Started

  • Overview
  • Installation

API Documentation

  • Core Library

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
  • Overview: module code

All modules for which code is available

  • jax_privacy.accounting
  • jax_privacy.auditing
  • jax_privacy.batch_selection
  • jax_privacy.clipping
  • jax_privacy.execution_plan
  • jax_privacy.experimental.compilation_utils
  • jax_privacy.experimental.monte_carlo.delta_calculation
  • jax_privacy.experimental.monte_carlo.sample_generation
  • jax_privacy.experimental.training
  • jax_privacy.keras_api
  • 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
  • jax_privacy.noise_addition
  • jax_privacy.optimizers

© Copyright 2025, Google DeepMind.

Built with Sphinx using a theme provided by Read the Docs.