NERSC / slurm-ray-cluster
☆55Updated last year
Alternatives and similar repositories for slurm-ray-cluster:
Users that are interested in slurm-ray-cluster are comparing it to the libraries listed below
- A brief tutorial on how to use ray/rllib/tun in slurm cluster.☆33Updated 4 years ago
- yaspi - Yet Another Slurm Python Interface☆44Updated 2 years ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Material for the SC22 Deep Learning at Scale Tutorial☆40Updated last year
- A simple Python wrapper for Slurm with flexibility in mind.☆137Updated last month
- Running Jax in PyTorch Lightning☆89Updated 3 months ago
- ☆112Updated last month
- SciML Benchmarking Suite for AI for Science☆39Updated 8 months ago
- MLBench Framework Core Python Library☆17Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- JMP is a Mixed Precision library for JAX.☆193Updated last month
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago
- Template repository for a Python 3-based data science project that uses Horovod.☆43Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated 2 years ago
- Distributed K-FAC Preconditioner for PyTorch☆85Updated this week
- Example of how to use Weights & Biases on Slurm☆113Updated 2 years ago
- ☆30Updated 6 months ago
- Normalizing Flows using JAX☆83Updated last year
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Lightning-like training API for JAX with Flax☆38Updated 3 months ago
- Instructions for setting up a SLURM cluster using Ubuntu 18.04.3 with GPUs.☆145Updated 4 years ago
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 2 years ago
- A Slurm dashboard for the terminal.☆84Updated last year
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆79Updated 4 years ago
- A functional training loops library for JAX☆86Updated last year
- PyTorch training at CSCS☆15Updated last year
- Train across all your devices, ezpz 🍋☆18Updated 3 weeks ago
- Worked example of the process from Python source to CUDA kernel execution with Numba☆37Updated 6 months ago