kvfrans / jax-vqvae-vqgan

Related projects ⓘ

Alternatives and complementary repositories for jax-vqvae-vqgan

• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated last week
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 2 years ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆32Updated last week
• chongyi-zheng / td_infonce
  ☆25Updated 11 months ago
• jianlanluo / SAQ
  ☆28Updated last year
• orybkin / lexa
  Discovering and Achieving Goals via World Models, NeurIPS 2021
  ☆83Updated 9 months ago
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆22Updated 2 weeks ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆21Updated 3 months ago
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆28Updated 3 months ago
• danijar / elements
  Building blocks for productive research
  ☆44Updated 2 weeks ago
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆52Updated 7 months ago
• google-deepmind / dmc_vision_benchmark
  ☆22Updated 4 months ago
• Michael-Beukman / RobocupGym
  Reinforcement Learning inside a 3D soccer simulation
  ☆24Updated last month
• FLAIROx / cultural-accumulation
  ☆12Updated 3 months ago
• danijar / ninjax
  General Modules for JAX
  ☆58Updated 3 months ago
• ademiadeniji / irm
  Intrinsic Reward Matching (IRM) implementation (from Adeniji and Xie et al 2022)
  ☆43Updated 9 months ago
• yuqingd / cusp
  ☆15Updated 2 years ago
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆25Updated last year
• seohongpark / PMA
  Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)
  ☆30Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆39Updated 2 years ago
• kvfrans / powderworld
  Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions
  ☆63Updated 2 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆25Updated last week
• DramaCow / jaxued
  ☆61Updated 2 months ago
• astanic / crafter-ood
  ☆20Updated last year
• vivekmyers / contrastive_planning
  Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"
  ☆36Updated 3 months ago
• stepjam / BPP
  Bingham Policy Parameterization for 3D Rotations in Reinforcement Learning
  ☆24Updated 2 years ago
• imgeorgiev / PWM
  PWM: Policy Learning with Large World Models
  ☆37Updated 2 months ago
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆22Updated 2 months ago
• quasimetric-learning / quasimetric-rl
  Open source code for paper "Optimal Goal-Reaching Reinforcement Learning via Quasimetric Learning" ICML 2023
  ☆44Updated 6 months ago
• schmidtdominik / LAPO
  Code for the ICLR 2024 spotlight paper: "Learning to Act without Actions" (introducing Latent Action Policies)
  ☆74Updated 3 months ago