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:
- ☆158Updated last year
- Tools for JAX☆49Updated 2 weeks ago
- ☆61Updated 2 years ago
- yaspi - Yet Another Slurm Python Interface☆45Updated 3 years ago
- Documentation:☆123Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Mathematical operations for JAX pytrees☆200Updated 10 months ago
- ☆80Updated 4 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 8 months ago
- Pytrees + dataclasses ❤️☆67Updated 5 months ago
- JAX Arrays for human consumption☆107Updated 3 months ago
- Exponential families for JAX☆75Updated 2 weeks ago
- Normalizing Flows using JAX☆84Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Bibtex for various Python science and machine learning software☆33Updated 2 years ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆97Updated 5 months ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- ☆51Updated 2 years ago
- Running Jax in PyTorch Lightning☆113Updated 9 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆496Updated 3 weeks ago
- ☆220Updated last week
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆15Updated 4 years ago
- Normalizing Flows in Jax☆107Updated 5 years ago
- A generic interface for linear algebra backends☆73Updated 7 months ago
- Bayesian algorithm execution (BAX)☆50Updated 4 years ago
- OpTree: Optimized PyTree Utilities☆196Updated this week
- Stencil computations in JAX☆71Updated 2 years ago
- A functional training loops library for JAX☆88Updated last year