Skip to main content
Ctrl+K

DiffuseNNX documentation

  • Installation
  • Quick Start
  • API Reference
  • Interfaces
  • Samplers
    • Configuration
    • Utilities
    • Contributing
    • Testing
    • FSDP In Jax NNX
  • Installation
  • Quick Start
  • API Reference
  • Interfaces
  • Samplers
  • Configuration
  • Utilities
  • Contributing
  • Testing
  • FSDP In Jax NNX

Section Navigation

API Documentation

  • Interfaces API
  • Samplers API
  • Evaluation API
  • Utilities API
  • API Reference

API Reference#

This section contains the complete API reference for the DiffuseNNX library.

API Documentation

  • Interfaces API
    • Continuous Interfaces
    • REPA Interfaces
  • Samplers API
    • Samplers
    • EulerSampler
    • EulerJumpSampler
    • HeunSampler
    • EulerMaruyamaSampler
  • Evaluation API
    • Frechet Inception Distance
    • Utility
  • Utilities API
    • Checkpointing
    • EMA (Exponential Moving Average)
    • Google Cloud Utilities
    • Sharding
    • Visualization

previous

Quick Start

next

Interfaces API

This Page

  • Show Source

© Copyright 2025, Nanye Ma.

Created using Sphinx 8.1.3.

Built with the PyData Sphinx Theme 0.16.1.