pengzhenghao / use-ray-with-slurm

Related projects ⓘ

Alternatives and complementary repositories for use-ray-with-slurm

• NERSC / slurm-ray-cluster
  ☆52Updated last year
• albanie / yaspi
  yaspi - Yet Another Slurm Python Interface
  ☆42Updated 2 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆43Updated this week
• jax-ml / jax-ai-stack
  ☆77Updated this week
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆88Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆41Updated last month
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆64Updated 4 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆187Updated 6 months ago
• google-deepmind / jaxline
  ☆152Updated 11 months ago
• cgarciae / simple-pytree
  A dead simple Python package for creating custom JAX pytree objects
  ☆62Updated last week
• google-deepmind / tf2jax
  ☆105Updated 2 weeks ago
• yuanqing-wang / galax
  Graph Learning with JAX
  ☆14Updated 2 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆48Updated 3 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆20Updated 4 months ago
• SymJAX / SymJAX
  Documentation:
  ☆120Updated last year
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆42Updated 4 years ago
• PredictiveIntelligenceLab / JAX-BO
  ☆30Updated 2 years ago
• kazukiosawa / ngd_in_wide_nn
  simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…
  ☆14Updated 4 years ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆105Updated last week
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• gehring / fax
  ☆78Updated 3 years ago
• KarenUllrich / binary-VAE
  A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.
  ☆21Updated 4 years ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆34Updated 6 months ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆55Updated this week
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆37Updated 4 years ago
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆10Updated 4 years ago
• amq92 / simple_slurm
  A simple Python wrapper for Slurm with flexibility in mind.
  ☆121Updated last week