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
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- yaspi - Yet Another Slurm Python Interface☆44Updated 2 years ago
- ☆112Updated last month
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆58Updated this week
- Material for the SC22 Deep Learning at Scale Tutorial☆40Updated last year
- A simple Python wrapper for Slurm with flexibility in mind.☆132Updated last month
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- SciML Benchmarking Suite for AI for Science☆39Updated 7 months ago
- JMP is a Mixed Precision library for JAX.☆193Updated last month
- ☆10Updated 2 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated 2 years ago
- High dimensional black-box optimizer using Latent Action Monte Carlo Tree Search algorithm☆27Updated 2 years ago
- A simple library for scaling up JAX programs☆134Updated 4 months ago
- Running Jax in PyTorch Lightning☆89Updated 2 months ago
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆20Updated 2 years ago
- Distributed K-FAC Preconditioner for PyTorch☆85Updated this week
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Instructions for setting up a SLURM cluster using Ubuntu 18.04.3 with GPUs.☆145Updated 4 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆11Updated 3 years ago
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 2 years ago
- Example of how to use Weights & Biases on Slurm☆113Updated 2 years ago
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆79Updated 4 years ago
- Worked example of the process from Python source to CUDA kernel execution with Numba☆37Updated 6 months ago
- Lightning-like training API for JAX with Flax☆38Updated 3 months ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Python implementation for Combining Latent Space and Structured Kernels for Bayesian Optimization over Combinatorial Spaces.☆13Updated 3 years ago
- Interpretating the latent space representations of attention head outputs for LLMs☆30Updated 7 months ago
- OpTree: Optimized PyTree Utilities☆171Updated this week