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

All modules for which code is available

  • jax_privacy.accounting.accountants
  • jax_privacy.accounting.analysis
  • jax_privacy.accounting.calibrate
  • jax_privacy.dp_sgd.gradients

© Copyright 2025, Google DeepMind.

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