NNTool¶
nntool is a package designed to provide seamless Python function execution on Slurm for machine learning research, with useful utilities for experiment tracking and management.
from nntool.slurm import SlurmConfig, slurm_fn
@slurm_fn
def run_on_slurm(a, b):
return a + b
slurm_config = SlurmConfig(
mode="slurm",
partition="PARTITION",
job_name="EXAMPLE",
tasks_per_node=1,
cpus_per_task=8,
mem="1GB",
)
job = run_on_slurm[slurm_config](1, b=2) # job is submitted to slurm
result = job.result() # block and get the result => 3
Installation¶
NNTool is tested and supported on the following systems:
Python 3.10-3.13
Linux systems with Slurm installed