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 3 years ago
- SciML Benchmarking Suite for AI for Science☆39Updated 9 months ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated last year
- A simple Python wrapper for Slurm with flexibility in mind.☆140Updated 2 months ago
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 2 years ago
- ☆114Updated last week
- Running Jax in PyTorch Lightning☆99Updated 4 months ago
- JAX Arrays for human consumption☆92Updated last year
- Train across all your devices, ezpz 🍋☆20Updated this week
- ☆156Updated last year
- JMP is a Mixed Precision library for JAX.☆196Updated 3 months ago
- Example of how to use Weights & Biases on Slurm☆114Updated 2 years ago
- Repository of machine learning benchmarks☆36Updated this week
- Distributed K-FAC preconditioner for PyTorch☆85Updated this week
- ☆43Updated last month
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆61Updated 2 months ago
- ☆30Updated 7 months ago
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆21Updated 3 years ago
- Distributed Bayesian Optimization☆23Updated 4 years ago
- SC23 Deep Learning at Scale Tutorial Material☆44Updated 7 months ago
- A simple library for scaling up JAX programs☆134Updated 6 months ago
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆120Updated last week
- ☆103Updated 10 months ago
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆79Updated 4 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆270Updated last week
- Instructions for setting up a SLURM cluster using Ubuntu 18.04.3 with GPUs.☆151Updated 4 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago