in scripts/run.py [0:0]
def _create_t5x_custom_job(
display_name: str,
machine_type: str,
accelerator_type: str,
accelerator_count: int,
image_uri: str,
run_mode: str,
gin_files: List[str],
model_dir: str,
gin_search_paths: Optional[List[str]] = None,
tfds_data_dir: Optional[str] = None,
replica_count: int = 1,
gin_overwrites: Optional[List[str]] = None,
base_output_dir: Optional[str] = None,