ezpz

Train across all your {NVIDIA, AMD, Intel, MPS, ...} accelerators, ezpz 🍋.

See 🍋 ezpz docs for additional information.

🐣 Getting Started

1. 🏖️ Setup environment¹ (see Shell Environment):

   source <(curl https://raw.githubusercontent.com/saforem2/ezpz/refs/heads/main/src/ezpz/bin/utils.sh) && ezpz_setup_env
   

2. 🐍 Install ezpz (see Python Library):

   python3 -m pip install "git+https://github.com/saforem2/ezpz"
   

3. 🚀 Launch any *.py² from python (see Launch):

   python3 -m ezpz.launch -m ezpz.test_dist
   

😎 2 ez.

---

1. This will 🪄 automagically source ezpz/bin/utils.sh and (&&) call ezpz_setup_env to setup your python environment. ↩

2. Technically, we're launching (-m ezpz.launch) the ezpz/test_dist.py as a module (-m), in this example. ↩