Alternatives and similar repositories for jax-vs-pytorch

Users that are interested in jax-vs-pytorch are comparing it to the libraries listed below

• ToruOwO / mimex
  PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"
  ☆16Updated 2 years ago
• filteredcophy / FilteredCoPhy
  ☆10Updated 2 years ago
• VITA-Group / SSM-Bottleneck
  [ICLR'25] "Understanding Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing" by Peihao Wang, Ruisi Cai, Yue…
  ☆13Updated 3 months ago
• yuqingd / cusp
  ☆15Updated 2 years ago
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆27Updated 3 years ago
• agopal42 / permakey
  Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…
  ☆9Updated 6 months ago
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆31Updated 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• siddharthverma314 / clcp-neurips-2020
  Code for Continual Learning of Control Primitives
  ☆18Updated 4 years ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆53Updated 7 months ago
• facebookresearch / gwil
  Cross-Domain Imitation Learning via Optimal Transport
  ☆25Updated 3 years ago
• shwinshaker / LipGrow
  An adaptive training algorithm for residual network
  ☆15Updated 4 years ago
• nirgreshler / bayesian-online-planning
  The code for the paper "A Bayesian Approach to Online Planning" published in ICML 2024.
  ☆11Updated last year
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 8 months ago
• ling-pan / GAFN
  ☆24Updated last year
• NingMiao / InstaAug
  ☆12Updated 2 years ago
• yilundu / ebm_compositionality
  [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models
  ☆45Updated 2 years ago
• subho406 / agalite
  AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)
  ☆19Updated 9 months ago
• Snagnar / Hieros
  Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper
  ☆18Updated last year
• facebookresearch / ExPLORe
  This is code to accompany the paper "Accelerating Exploration with Unlabeled Prior Data".
  ☆24Updated last year
• ComputationalRobotics / TRAC
  This repository is the official implementation of the TRAC optimizer in Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement …
  ☆28Updated 2 months ago
• yiqiwang8177 / Official-codebase-for-Decision-Transducer
  This is the pytorch implementation of the UAI2023 paper "A Trajectory is Worth Three Sentences: Multimodal Transformer for Offline Reinf…
  ☆11Updated last year
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 3 years ago
• maxreciprocate / offline
  Offline RL experiments
  ☆15Updated 2 years ago
• rujiewu / Bongard-OpenWorld
  This is the official code implementation of Bongard-OpenWorld (ICLR 2024).
  ☆12Updated 6 months ago
• allenai / ask4help
  Code for the Ask4Help project
  ☆22Updated 2 years ago
• yilundu / comet
  [NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts
  ☆62Updated 2 years ago
• facebookresearch / ssorl
  Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories
  ☆42Updated 2 years ago
• bdusell / stack-attention
  Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"
  ☆17Updated last year
• marc-rigter / waker
  Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.
  ☆27Updated last year