Alternatives and similar repositories for HesScale

Users that are interested in HesScale are comparing it to the libraries listed below

• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 3 years ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆29Updated last year
• gcucurull / maml_flax
  Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.
  ☆20Updated 5 years ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆41Updated 7 months ago
• ivandariojr / LyapunovLearning
  ☆46Updated 2 years ago
• FLAIROx / popjym
  POPGym Library in JAX
  ☆12Updated last year
• facebookresearch / ssorl
  Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories
  ☆42Updated 2 years ago
• facebookresearch / how-to-autorl
  Plug-and-play hydra sweepers for the EA-based multifidelity method DEHB and several population-based training variations, all proven to e…
  ☆85Updated 2 years ago
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆46Updated 3 years ago
• zdhNarsil / Stochastic-Marginal-Actor-Critic
  Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".
  ☆24Updated 2 years ago
• twitter-research / hyperbolic-rl
  ☆58Updated 3 years ago
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 3 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆52Updated 2 years ago
• xingchenwan / bgpbt
  [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)
  ☆29Updated 3 years ago
• neuroevobench / neuroevobench
  Neuroevolution Benchmark in JAX 🦕
  ☆42Updated 2 years ago
• lili-chen / SEER
  Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.
  ☆21Updated 4 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated last year
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆112Updated 2 years ago
• ElisevanderPol / symmetrizer
  ☆32Updated 4 years ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆32Updated 2 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 4 years ago
• Jiacheng-Zhu-AIML / WGPOT
  The Wasserstein Distance and Optimal Transport Map of Gaussian Processes
  ☆52Updated 5 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆45Updated last year
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆37Updated 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 4 years ago
• rail-berkeley / design-bench
  ☆53Updated 3 years ago
• google-deepmind / affordances_option_models
  ☆23Updated 4 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆16Updated 2 years ago