Alternatives and similar repositories for hydra-ray-demo

Users that are interested in hydra-ray-demo are comparing it to the libraries listed below

• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆63Updated 2 months ago
• google / deluca
  Performant, differentiable reinforcement learning
  ☆121Updated this week
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆32Updated 2 years ago
• ElisevanderPol / PRAE
  Plannable Approximations to MDP Homomorphisms: Equivariance under Actions
  ☆30Updated 4 years ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• jhejna / morphology-opt
  Code for the paper Task Agnostic Morphology Evolution.
  ☆20Updated 4 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆112Updated 3 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆54Updated last month
• madupite / madupite
  a High-Performance Distributed Solver for Large-Scale Markov Decision Processes (MDP) relying on Inexact Policy Iteration; for Python and…
  ☆25Updated 2 months ago
• cognizant-ai-labs / autoinit
  AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks
  ☆28Updated 2 years ago
• google-research / robodesk
  ☆81Updated 2 years ago
• google-deepmind / zipfian_environments
  ☆28Updated 2 years ago
• google-deepmind / agent_debugger
  Causal Analysis of Agent Behavior for AI Safety
  ☆18Updated 2 years ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆114Updated this week
• sergio-hcsoft / FractalAI
  Cellular automaton-based calculus for the masses
  ☆24Updated 6 years ago
• jeguzzi / mighty-thymio
  Complementary material to EAAAI18 Paper "Mighty Thymio for Higher-Level Robotics Education"
  ☆19Updated 2 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆43Updated 4 years ago
• newera-001 / motor-system
  A project copied from google-research which named motion-imitation was rewrited with PyTorch
  ☆9Updated 2 years ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• orybkin / lexa-benchmark
  ☆42Updated 3 years ago
• google-deepmind / tf2jax
  ☆114Updated this week
• google / cloud-berg
  Berg - Run GPU-backed experiments on gcloud
  ☆26Updated 6 years ago
• vtopt / qnstop
  Quasi-Newton Algorithm for Stochastic Optimization
  ☆10Updated 3 years ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆37Updated last month