Fix warnings with -O3 in jax-0.6.2

[Flamefire]

(created using eb --new-pr)

[Flamefire]

Test report by @Flamefire
SUCCESS
Build succeeded for 1 out of 1 (total: 39 mins 32 secs) (1 easyconfigs in total)
i8019 - Linux Rocky Linux 9.6, x86_64, AMD EPYC 7352 24-Core Processor (zen2), 8 x NVIDIA NVIDIA A100-SXM4-40GB, 580.65.06, Python 3.9.21
See https://gist.github.com/Flamefire/c8bc5b2e388f5371d9fe2d1b1f3c2313 for a full test report.

[branfosj]

@boegelbot please test @ jsc-zen3

[branfosj]

Test report by @branfosj
SUCCESS
Build succeeded for 1 out of 1 (total: 44 mins 39 secs) (1 easyconfigs in total)
bear-pg0210u03b - Linux RHEL 8.10, x86_64, Intel(R) Xeon(R) Platinum 8480CL (sapphirerapids), Python 3.6.8
See https://gist.github.com/branfosj/8b24be066c2a5adddafe55a4f7a43212 for a full test report.

[boegelbot]

@branfosj: Request for testing this PR well received on jsczen3l1.int.jsc-zen3.fz-juelich.de

PR test command 'if [[ develop != 'develop' ]]; then EB_BRANCH=develop ./easybuild_develop.sh 2> /dev/null 1>&2; EB_PREFIX=/home/boegelbot/easybuild/develop source init_env_easybuild_develop.sh; fi; EB_PR=25207 EB_ARGS= EB_CONTAINER= EB_REPO=easybuild-easyconfigs EB_BRANCH=develop /opt/software/slurm/bin/sbatch --job-name test_PR_25207 --ntasks=8 ~/boegelbot/eb_from_pr_upload_jsc-zen3.sh' executed!

• exit code: 0
• output:

Submitted batch job 9560

Test results coming soon (I hope)...

Details

- notification for comment with ID 3846854553 processed

Message to humans: this is just bookkeeping information for me,
it is of no use to you (unless you think I have a bug, which I don't).

[jfgrimm]

Test report by @jfgrimm
SUCCESS
Build succeeded for 1 out of 1 (total: 1 hour 27 mins 43 secs) (1 easyconfigs in total)
himem02.viking2.yor.alces.network - Linux Rocky Linux 8.10, x86_64, AMD EPYC 7643 48-Core Processor, Python 3.6.8
See https://gist.github.com/jfgrimm/0c9d8f4c13bb4d0388299e28f23870b0 for a full test report.

[boegelbot]

Test report by @boegelbot
SUCCESS
Build succeeded for 1 out of 1 (total: 9 hours 38 mins 52 secs) (1 easyconfigs in total)
jsczen3c3.int.jsc-zen3.fz-juelich.de - Linux Rocky Linux 9.7, x86_64, AMD EPYC-Milan Processor (zen3), Python 3.9.23
See https://gist.github.com/boegelbot/1e72b8e39a69bd513c9580fe4d0c5c45 for a full test report.

[branfosj]

Going in, thanks @Flamefire!

[Flamefire]

Test report by @Flamefire
SUCCESS
Build succeeded for 1 out of 1 (total: 26 mins 34 secs) (1 easyconfigs in total)
c108 - Linux Rocky Linux 9.6, x86_64, AMD EPYC 9334 32-Core Processor (zen4), 4 x NVIDIA NVIDIA H100, 580.65.06, Python 3.9.21
See https://gist.github.com/Flamefire/bc4bd105f4197c88d58fcd3d7b34c867 for a full test report.