pengzhenghao / use-ray-with-slurmLinks
A brief tutorial on how to use ray/rllib/tun in slurm cluster.
☆33Updated 4 years ago
Alternatives and similar repositories for use-ray-with-slurm
Users that are interested in use-ray-with-slurm are comparing it to the libraries listed below
Sorting:
- ☆59Updated 2 years ago
- ☆158Updated last year
- yaspi - Yet Another Slurm Python Interface☆44Updated 3 years ago
- ☆80Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A Python Toolkit for Managing a Large Number of Experiments☆32Updated last year
- Bayesian algorithm execution (BAX)☆50Updated 3 years ago
- Tools for JAX☆49Updated last week
- Documentation:☆121Updated 2 years ago
- Pytrees + dataclasses ❤️☆67Updated 3 months ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- JAX Arrays for human consumption☆106Updated last month
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 2 weeks ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 6 months ago
- ☆203Updated this week
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 3 months ago
- ☆115Updated last week
- Mathematical operations for JAX pytrees☆199Updated 8 months ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆33Updated 3 months ago
- A simple Python wrapper for Slurm with flexibility in mind.☆150Updated 2 months ago
- Tidy autoregressive inference in JAX☆14Updated last week
- Jax SSM Library☆49Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆40Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Normalizing Flows using JAX☆84Updated last year
- Exponential families for JAX☆73Updated last week
- Normalizing Flows in Jax☆107Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- 🌳 Python implementation of single-player Monte-Carlo Tree Search.☆63Updated 4 years ago
- ☆68Updated last year