release_images_training.yml

---
release_images:
  1:
    framework: "huggingface_pytorch"
    version: "1.13.1"
    arch_type: "x86"
    hf_transformers: "4.36.2"
    training:
      device_types: [ "neuronx" ]
      python_versions: [ "py310" ]
      os_version: "ubuntu20.04"
      neuron_sdk_version: "sdk2.18.0"
      example: False
      disable_sm_tag: True
      force_release: False
  2:
    framework: "huggingface_pytorch"
    version: "1.13.1"
    arch_type: "x86"
    hf_transformers: "4.26.0"
    training:
      device_types: [ "gpu" ]
      python_versions: [ "py39" ]
      os_version: "ubuntu20.04"
      cuda_version: "cu117"
      example: False
      disable_sm_tag: False
      force_release: False
  3:
    framework: "huggingface_pytorch"
    version: "2.1.0"
    hf_transformers: "4.36.0"
    arch_type: "x86"
    training:
      device_types: ["gpu"]
      python_versions: [ "py310" ]
      os_version: "ubuntu20.04"
      cuda_version: "cu121"
      example: False
      disable_sm_tag: False
      force_release: False
  4:
    framework: "huggingface_pytorch"
    version: "2.1.2"
    arch_type: "x86"
    hf_transformers: "4.43.2"
    training:
      device_types: [ "neuronx" ]
      python_versions: [ "py310" ]
      os_version: "ubuntu20.04"
      neuron_sdk_version: "sdk2.20.0"
      example: False
      disable_sm_tag: True
      force_release: False
  5:
    framework: "huggingface_pytorch"
    version: "2.3.0"
    hf_transformers: "4.46.1"
    arch_type: "x86"
    training:
      device_types: ["gpu"]
      python_versions: [ "py311" ]
      os_version: "ubuntu20.04"
      cuda_version: "cu121"
      example: False
      disable_sm_tag: False
      force_release: False
  6:
    framework: "autogluon"
    version: "1.2.0"
    arch_type: "x86"
    training:
      device_types: ["cpu", "gpu"]
      python_versions: ["py311"]
      os_version: "ubuntu22.04"
      cuda_version: "cu124"
      example: False
      disable_sm_tag: False
      force_release: False