【Bug】test_lm_score_may_fail_numerically_for_external_meliad

# Env:
python 3.10
A100 40GB
Cuda compilation tools, release 12.3, V12.3.103
Build cuda_12.3.r12.3/compiler.33492891_0

# Test:
```
bash run_tests.sh
```

# Log:
```
[ OK ] LmInferenceTest.test_lm_decode
[ RUN ] LmInferenceTest.test_lm_score_may_fail_numerically_for_external_meliad
[ FAILED ] LmInferenceTest.test_lm_score_may_fail_numerically_for_external_meliad
======================================================================
FAIL: test_lm_score_may_fail_numerically_for_external_meliad (__main__.LmInferenceTest)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/workspace/alphageometry/lm_inference_test.py", line 82, in test_lm_score_may_fail_numerically_for_external_meliad
self.assertEqual(
AssertionError: Lists differ: [-1.1633697, -1.122621] != [-1.1860729455947876, -1.1022869348526]

First differing element 0:
-1.1633697
-1.1860729455947876

- [-1.1633697, -1.122621]
+ [-1.1860729455947876, -1.1022869348526]

----------------------------------------------------------------------
Ran 2 tests in 133.525s

FAILED (failures=1)
```