Skip to main content

Convert SLURM submit scripts to Run.ai configurations using AI

Project description

s2r - SLURM to Run.ai Converter

Convert SLURM batch scripts to Run.ai configurations using AI.

Installation

pip install s2r

Quick Start

CLI Usage

# Convert file: saves job.yaml, prints runai CLI command to stdout
s2r job.slurm

# Convert file: saves output.yaml only (no CLI output)
s2r job.slurm output.yaml

# Convert from stdin: prints runai CLI command to stdout (no file)
s2r < my_slurm_script.sh

Library Usage

from s2r import convert_slurm_to_runai

slurm_script = """
#!/bin/bash
#SBATCH --job-name=my-job
#SBATCH --gres=gpu:2
#SBATCH --mem=32G

python train.py
"""

runai_config = convert_slurm_to_runai(slurm_script)
print(runai_config)

How It Works

  1. Client: The s2r library signs your SLURM script with HMAC-SHA256
  2. API: Sends the signed request to an AWS Lambda endpoint
  3. AI: Lambda calls AWS Bedrock (Claude) to perform the conversion
  4. Response: Returns the Run.ai YAML configuration or CLI commands

Features

  • Free to use: The service is provided at no cost (rate-limited)
  • Secure: Signed requests prevent unauthorized API usage
  • Rate-limited: 100 requests per IP per day
  • Simple: Works with stdin, files, or as a library

Configuration

By default, the tool uses a public API endpoint. If you're deploying your own:

export S2R_API_ENDPOINT=https://your-lambda-url.lambda-url.us-east-1.on.aws/

Example

Given this SLURM script:

#!/bin/bash
#SBATCH --job-name=pytorch-training
#SBATCH --nodes=1
#SBATCH --cpus-per-task=8
#SBATCH --mem=32G
#SBATCH --gres=gpu:2
#SBATCH --time=24:00:00

python train.py --epochs 100

The tool will generate an equivalent Run.ai configuration with:

  • GPU resource requests (2 GPUs)
  • CPU and memory allocations
  • Job name and command
  • Appropriate container/image specifications

Documentation

Current Deployment Status

Deployed Lambda Function (us-west-2):

  • Model: Claude Sonnet 4.5 (20250929)
  • Rate Limit: 100 requests/IP/day

Known Issues:

  • Function URL may return 403 Forbidden due to IAM permission constraints
  • See troubleshooting guide for resolution

Development

See CLAUDE.md for development commands and quick reference.

For detailed architecture and deployment information, see the docs/ directory.

Self-Hosting

To deploy your own instance:

# Clone repository
git clone https://github.com/yourusername/slurm2runai.git
cd slurm2runai

# Deploy Lambda function
cd lambda
python3 -m zipfile -c lambda.zip lambda_function.py
aws lambda create-function --function-name s2r-converter ...

# See https://github.com/dirkpetersen/slurm2runai/blob/main/docs/deployment.md for full instructions

License

MIT License - see LICENSE file for details.

Contributing

Contributions welcome! Please:

  1. Open an issue to discuss changes
  2. Follow the code style (ruff)
  3. Add tests for new features
  4. Update documentation

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

s2r-0.2.1.tar.gz (10.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

s2r-0.2.1-py3-none-any.whl (9.2 kB view details)

Uploaded Python 3

File details

Details for the file s2r-0.2.1.tar.gz.

File metadata

  • Download URL: s2r-0.2.1.tar.gz
  • Upload date:
  • Size: 10.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for s2r-0.2.1.tar.gz
Algorithm Hash digest
SHA256 f09b95740c1eb9231c5bc6451b9139d89d52bc073ea444301e322993a16c6f8c
MD5 e0e1c23d103c6413494324b044c38492
BLAKE2b-256 d7ec4bd98f72d37d73b328a1fdb93b2b90682cce0161b1af5609c8cab18c62c0

See more details on using hashes here.

Provenance

The following attestation bundles were made for s2r-0.2.1.tar.gz:

Publisher: publish-pypi.yml on dirkpetersen/slurm2runai

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file s2r-0.2.1-py3-none-any.whl.

File metadata

  • Download URL: s2r-0.2.1-py3-none-any.whl
  • Upload date:
  • Size: 9.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for s2r-0.2.1-py3-none-any.whl
Algorithm Hash digest
SHA256 62a4b29112318fd15734854b0e7193a13948e9c8798aa8a40ac4ff21870cf4b6
MD5 828856afec5131aa7cbd2eeba5231f19
BLAKE2b-256 a465b2844836dd0517e664c6a35b92b3b35bd925642500e464e5106ed2281917

See more details on using hashes here.

Provenance

The following attestation bundles were made for s2r-0.2.1-py3-none-any.whl:

Publisher: publish-pypi.yml on dirkpetersen/slurm2runai

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page