Alternatives and similar repositories for contrastive-rl

Users that are interested in contrastive-rl are comparing it to the libraries listed below

• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• lucidrains / vit-arc-slot
  Explorations into improving ViTArc with Slot Attention
  ☆43Updated last year
• lucidrains / improving-transformers-world-model-for-rl
  Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch
  ☆150Updated 8 months ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆71Updated 2 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆107Updated last year
• lucidrains / x-transformers-rl
  Implementation of a transformer for reinforcement learning using `x-transformers`
  ☆72Updated 3 months ago
• milmor / diffusion-transformer
  Implementation of Diffusion Transformer Model in Pytorch
  ☆73Updated 2 weeks ago
• nreHieW / minARImageGen
  Autoregressive Image Generation
  ☆31Updated 7 months ago
• lucidrains / lbm-training-framework
  Training framework for Large Behavioral Models
  ☆27Updated 4 months ago
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆103Updated 3 months ago
• MambaMixer / M2
  ☆48Updated last year
• lucidrains / deep-cross-attention
  Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch
  ☆96Updated 10 months ago
• lucidrains / sinkhorn-router-pytorch
  Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise
  ☆40Updated last year
• lucidrains / agent-attention-pytorch
  Implementation of Agent Attention in Pytorch
  ☆93Updated last year
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆112Updated last year
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆57Updated last year
• galilai-group / llm-jepa
  ☆158Updated 3 months ago
• lucidrains / mixture-of-attention
  Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts
  ☆122Updated last year
• facebookresearch / Mixture-of-Transformers
  Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Models. TMLR 2025.
  ☆138Updated 4 months ago
• zhixuan-lin / forgetting-transformer
  [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning
  ☆137Updated last month
• lucidrains / LVMAE-pytorch
  Implementation of the proposed LVMAE, from the paper, Extending Video Masked Autoencoders to 128 frames, in Pytorch
  ☆55Updated last year
• Shalev-Lifshitz / MultiAgentVerification
  Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers
  ☆26Updated 10 months ago
• lucidrains / h-net-dynamic-chunking
  Implementation of the dynamic chunking mechanism in H-net by Hwang et al. of Carnegie Mellon
  ☆66Updated last week
• raunaqbhirangi / hiss
  Hierarchical State Space Models
  ☆48Updated last year
• lucidrains / x-evolution
  Implementation of various evolutionary algorithms, starting with evolutionary strategies
  ☆43Updated 3 weeks ago
• Weixin-Liang / Mixture-of-Mamba
  ☆50Updated 11 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆90Updated last year
• ethanbar11 / ssm_2d
  More dimensions = More fun
  ☆26Updated last year
• minhoooo1 / CatMAE
  CatMAE
  ☆14Updated 2 years ago
• philippe-eecs / small-vision
  A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.
  ☆34Updated last year