[Python] unable to install autotrain-advanced in sagemaker ai

I'm trying to install autotrain-advanced python library to train ollama llama model in sagemaker ai but i'm getting 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)

Continue reading...