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:
- ☆55Updated last year
- ☆80Updated 3 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- yaspi - Yet Another Slurm Python Interface☆44Updated 3 years ago
- A collection of graph neural networks implementations in JAX☆32Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 4 years ago
- ☆15Updated 4 years ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 6 months ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- JAX Arrays for human consumption☆91Updated last year
- Documentation:☆120Updated 2 years ago
- Pytrees + dataclasses ❤️☆63Updated last month
- Truncated Normal Distribution in PyTorch☆84Updated last year
- Exponential families for JAX☆69Updated last week
- Tools for JAX☆47Updated this week
- diffusionjax is a simple and accessible diffusion models package in JAX☆45Updated 4 months ago
- A Python Toolkit for Managing a Large Number of Experiments☆32Updated last year
- ☆52Updated 2 years ago
- A generic interface for linear algebra backends☆73Updated 3 months ago
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- ☆114Updated this week
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- Example of how to use Weights & Biases on Slurm☆115Updated 2 years ago
- Reduce multiple PyTorch TensorBoard runs to new event (or CSV) files.☆73Updated 2 weeks ago
- WandB sweeps integration with Hydra sweeper☆49Updated last year
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Updated last year
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆21Updated 4 years ago
- RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation☆8Updated 3 years ago