NERSC / slurm-ray-clusterLinks
☆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
Sorting:
- yaspi - Yet Another Slurm Python Interface☆44Updated 3 years ago
- A brief tutorial on how to use ray/rllib/tun in slurm cluster.☆33Updated 4 years ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated last year
- SciML Benchmarking Suite for AI for Science☆41Updated 10 months ago
- A simple Python wrapper for Slurm with flexibility in mind.☆142Updated 3 weeks ago
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆61Updated 2 months ago
- ☆114Updated this week
- JMP is a Mixed Precision library for JAX.☆199Updated 4 months ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Example of how to use Weights & Biases on Slurm☆115Updated 2 years ago
- ☆43Updated this week
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆79Updated 4 years ago
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated 2 years ago
- Running Jax in PyTorch Lightning☆101Updated 5 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- OpTree: Optimized PyTree Utilities☆185Updated this week
- A functional training loops library for JAX☆88Updated last year
- A simple library for scaling up JAX programs☆137Updated 7 months ago
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 2 years ago
- ☆104Updated 11 months ago
- JAX Arrays for human consumption☆91Updated last year
- Repository of machine learning benchmarks☆36Updated this week
- Multiple dispatch over abstract array types in JAX.☆124Updated this week
- A Python package of computer vision models for the Equinox ecosystem.☆104Updated 10 months ago
- ☆157Updated last year
- Graph neural networks in JAX.☆67Updated 11 months ago
- SC23 Deep Learning at Scale Tutorial Material☆45Updated 8 months ago
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆21Updated 3 years ago
- Collection of small examples for running on ALCF resources☆19Updated last week