class MLEngineBackend:
    ml_engine_args_flag = "ml_engine"

    ml_engine_backend_torch = "torch"
    ml_engine_backend_tf = "tf"
    ml_engine_backend_jax = "jax"
    ml_engine_backend_mxnet = "mxnet"

    ml_device_type_gpu = "gpu"
    ml_device_type_cpu = "cpu"
    ml_device_type_mps = "mps"
