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