Update timeout of AzureML tests and add explicit runner in Ubuntu 24.04

[miguelgfierro]

Description

I want to limit each of the computations to 2h, but not all the tests of 1 category. For example, each test process should take less than 2h, but the whole GPU tests can take longer because maybe they are waiting for a free VM.

I used https://docs.github.com/en/actions/writing-workflows/workflow-syntax-for-github-actions#jobsjob_idtimeout-minutes but if it doesn't work I would need to do the timeout within the steps: https://docs.github.com/en/actions/writing-workflows/workflow-syntax-for-github-actions#jobsjob_idstepstimeout-minutes

Related Issues

Related to this discussion: #2189 (comment)

References

Checklist:

• I have followed the contribution guidelines and code style for this project.
• I have added tests covering my contributions.
• I have updated the documentation accordingly.
• I have signed the commits, e.g. git commit -s -m "your commit message".
• This PR is being made to staging branch AND NOT TO main branch.