Jax Config

Module: generative_models.core.jax_config

Source: generative_models/core/jax_config.py

Overview

JAX configuration optimization for maximum performance.

Functions

configure_jax

def configure_jax()

Module Statistics

• Classes: 0
• Functions: 1
• Imports: 5