Skip to content

Apptainer "kitchen sink" ML environment using Poetry

Notifications You must be signed in to change notification settings

PythonNut/apptainer-jax

Repository files navigation

apptainer-jax

Apptainer "kitchen sink" ML environment using Poetry

Building a new container with Slurm

sbatch cuda-jax.slurm

Running the container

cuda-jax-wrapper.sh <program> <arg1> <arg2> ...

You may want to give cuda-jax-wrapper.sh a convenient name (and put it in your PATH).

About

Apptainer "kitchen sink" ML environment using Poetry

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages