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
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆36Updated last year
• twitter-research / hyperbolic-rl
  ☆56Updated 2 years ago
• gcucurull / maml_flax
  Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.
  ☆20Updated 4 years ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆110Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 10 months ago
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• ivandariojr / LyapunovLearning
  ☆43Updated 2 years ago
• neuroevobench / neuroevobench
  Neuroevolution Benchmark in JAX 🦕
  ☆39Updated last year
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated last year
• CLAIRE-Labo / no-representation-no-trust
  Codebase to fully reproduce the results of "No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPO" (M…
  ☆27Updated 9 months ago
• xingchenwan / bgpbt
  [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)
  ☆28Updated 2 years ago
• facebookresearch / ssorl
  Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories
  ☆42Updated 2 years ago
• Jiacheng-Zhu-AIML / WGPOT
  The Wasserstein Distance and Optimal Transport Map of Gaussian Processes
  ☆53Updated 5 years ago
• lili-chen / SEER
  Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.
  ☆20Updated 4 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆34Updated last year
• subho406 / Recurrent-PPO-Jax
  Implementation of Proximal Policy Optimization in Jax+Flax
  ☆20Updated 2 years ago
• google-deepmind / tell_me_why_explanations_rl
  ☆37Updated 2 years ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated last year
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆39Updated 3 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆43Updated 2 years ago
• FLAIROx / popjym
  POPGym Library in JAX
  ☆11Updated last year
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 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
• danijar / elements
  Building blocks for productive research
  ☆59Updated 3 weeks 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…
  ☆83Updated last year
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 4 years ago
• radarFudan / mamba-minimal-jax
  ☆32Updated 9 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆87Updated last year