NERSC / slurm-ray-cluster
☆54Updated 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
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- A simple Python wrapper for Slurm with flexibility in mind.☆128Updated this week
- ☆111Updated last week
- SciML Benchmarking Suite for AI for Science☆39Updated 6 months ago
- Material for the SC22 Deep Learning at Scale Tutorial☆40Updated last year
- High dimensional black-box optimizer using Latent Action Monte Carlo Tree Search algorithm☆26Updated 2 years ago
- JMP is a Mixed Precision library for JAX.☆191Updated 2 weeks ago
- Example of how to use Weights & Biases on Slurm☆113Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆54Updated this week
- ☆28Updated 4 months ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- JAX Arrays for human consumption☆90Updated 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
- Documentation:☆119Updated last year
- Train across all your devices, ezpz 🍋☆18Updated this week
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 2 years ago
- ☆153Updated last year
- Instructions for setting up a SLURM cluster using Ubuntu 18.04.3 with GPUs.☆144Updated 4 years ago
- Distributed K-FAC Preconditioner for PyTorch☆85Updated this week
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆60Updated last week
- Running Jax in PyTorch Lightning☆86Updated 2 months ago
- A simple library for scaling up JAX programs☆129Updated 3 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆43Updated last month
- A Machine Learning workflow for Slurm.☆149Updated 4 years ago