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
  • Search

© Copyright 2025, Google DeepMind.

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