Error when installing autotrain-advanced Python library on Sagemaker AI

Question

I'm trying to install the autotrain-advanced Python library to train ollama Llama model in Sagemaker AI, but I'm getting an error.

This is my sagemaker instance configs:

instance type: ml.g4dn.12xlarge, Platform identifier:Amazon Linux 2, Jupyter Lab 3 (notebook-al2-v2)

Below is the full error:

    include/simsimd/probability.h:586:5: error: unknown type name ‘__m512h’; did you mean ‘__m512d’?
           __m512h log_ratio_b_vec = _simsimd_log2_f16_sapphire(ratio_b_vec);
           ^~~~~~~
           __m512d
      In file included from include/simsimd/simsimd.h:111:0,
                       from c/lib.c:54:
      include/simsimd/sparse.h: At top level:
      include/simsimd/sparse.h:277:9: error: attribute(target("avx512vbmi2")) is unknown
       #pragma GCC target("avx2", "avx512f", "avx512vl", "bmi2", "lzcnt", "popcnt", "avx512bw", "avx512vbmi2")
               ^~~
      include/simsimd/sparse.h: In function ‘_simsimd_intersect_u16x32_ice’:
      include/simsimd/sparse.h:294:19: error: incompatible types when initializing type ‘__m512i {aka __vector(8) long long int}’ using type ‘int’
           __m512i b01 = _mm512_shrdi_epi32(b, b, 16);
                         ^~~~~~~~~~~~~~~~~~
      include/simsimd/sparse.h:295:19: error: incompatible types when initializing type ‘__m512i {aka __vector(8) long long int}’ using type ‘int’
           __m512i b11 = _mm512_shrdi_epi32(b1, b1, 16);
                         ^~~~~~~~~~~~~~~~~~
      include/simsimd/sparse.h:296:19: error: incompatible types when initializing type ‘__m512i {aka __vector(8) long long int}’ using type ‘int’
           __m512i b21 = _mm512_shrdi_epi32(b2, b2, 16);
                         ^~~~~~~~~~~~~~~~~~
      include/simsimd/sparse.h:297:19: error: incompatible types when initializing type ‘__m512i {aka __vector(8) long long int}’ using type ‘int’
           __m512i b31 = _mm512_shrdi_epi32(b3, b3, 16);
                         ^~~~~~~~~~~~~~~~~~
      include/simsimd/sparse.h: At top level:
      include/simsimd/sparse.h:529:9: error: attribute(target("avx512vbmi2")) is unknown
       #pragma GCC target("avx2", "avx512f", "avx512vl", "bmi2", "lzcnt", "popcnt", "avx512bw", "avx512vbmi2", "avx512bf16", \
               ^~~
      include/simsimd/sparse.h:529:9: error: attribute(target("avx512bf16")) is unknown
      include/simsimd/sparse.h:529:9: error: attribute(target("avx512vnni")) is unknown
      include/simsimd/sparse.h:529:9: error: attribute(target("avx512vp2intersect")) is unknown
      include/simsimd/sparse.h: In function ‘simsimd_spdot_weights_u16_turin’:
      include/simsimd/sparse.h:739:27: error: incompatible types when assigning to type ‘__m256i {aka __vector(4) long long int}’ from type ‘int’
                   a_weights_vec = _mm256_maskz_compress_epi16(a_matches_any_in_b, a_weights_vec);
                                 ^
      include/simsimd/sparse.h:741:27: error: incompatible types when assigning to type ‘__m256i {aka __vector(4) long long int}’ from type ‘int’
                   b_weights_vec = _mm256_maskz_compress_epi16(b_matches_any_in_a, b_weights_vec);
                                 ^
      include/simsimd/sparse.h:742:70: error: ‘__m256bh’ undeclared (first use in this function); did you mean ‘__m256_u’?
                   product_vec.ymmps = _mm256_dpbf16_ps(product_vec.ymmps, (__m256bh)a_weights_vec, (__m256bh)b_weights_vec);
                                                                            ^~~~~~~~
                                                                            __m256_u
      include/simsimd/sparse.h:742:79: error: expected ‘)’ before ‘a_weights_vec’
                   product_vec.ymmps = _mm256_dpbf16_ps(product_vec.ymmps, (__m256bh)a_weights_vec, (__m256bh)b_weights_vec);
                                                                                     ^~~~~~~~~~~~~
      include/simsimd/sparse.h: In function ‘simsimd_spdot_counts_u16_turin’:
      include/simsimd/sparse.h:824:27: error: incompatible types when assigning to type ‘__m256i {aka __vector(4) long long int}’ from type ‘int’
                   a_weights_vec = _mm256_maskz_compress_epi16(a_matches_any_in_b, a_weights_vec);
                                 ^
      include/simsimd/sparse.h:826:27: error: incompatible types when assigning to type ‘__m256i {aka __vector(4) long long int}’ from type ‘int’
                   b_weights_vec = _mm256_maskz_compress_epi16(b_matches_any_in_a, b_weights_vec);
                                 ^
      include/simsimd/sparse.h:827:29: error: incompatible types when assigning to type ‘__m256i {aka __vector(4) long long int}’ from type ‘int’
                   product_vec.ymm = _mm256_dpwssds_epi32(product_vec.ymm, a_weights_vec, b_weights_vec);
                                   ^
      In file included from include/simsimd/curved.h:41:0,
                       from include/simsimd/simsimd.h:106,
                       from c/lib.c:54:
      include/simsimd/spatial.h: In function ‘_simsimd_substract_bf16x32_genoa’:
      include/simsimd/spatial.h:1676:26: note: The ABI for passing parameters with 64-byte alignment has changed in GCC 4.6
       SIMSIMD_INTERNAL __m512i _simsimd_substract_bf16x32_genoa(__m512i a_i16, __m512i b_i16) {
                                ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
      error: command '/usr/bin/gcc' failed with exit code 1
      [end of output]
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
  ERROR: Failed building wheel for simsimd
Successfully built ipadic pycocotools rouge-score seqeval inflate64 pyppmd
Failed to build sentencepiece simsimd
ERROR: Failed to build installable wheels for some pyproject.toml based projects (sentencepiece, simsimd)