Device

Module: generative_models.utils.jax.device

Source: generative_models/utils/jax/device.py

Overview

Device utilities for Workshop generative models.

This module provides a clean interface to the comprehensive device management system. All functionality has been moved to the core device management architecture.

Functions

get_recommended_batch_size

def get_recommended_batch_size()

setup_device_for_training

def setup_device_for_training()

verify_device_setup

def verify_device_setup()

Module Statistics

• Classes: 0
• Functions: 3
• Imports: 2