Alternatives and similar repositories for jax-vs-pytorch:

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

• yilundu / ebm_compositionality
  [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models
  ☆44Updated 2 years ago
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆27Updated 3 years ago
• ToruOwO / mimex
  PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"
  ☆16Updated last year
• facebookresearch / gwil
  Cross-Domain Imitation Learning via Optimal Transport
  ☆24Updated 2 years ago
• ruotianluo / lazy_related_work
  ☆14Updated 2 years ago
• ling-pan / GAFN
  ☆24Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆32Updated 2 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 4 months ago
• filteredcophy / FilteredCoPhy
  ☆10Updated 2 years ago
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated 8 months ago
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago
• zhxieml / PDT
  Implementation of ICML 2023 paper: Future-conditioned Unsupervised Pretraining for Decision Transformer
  ☆27Updated last year
• belindal / LaMPP
  Code for LaMPP: Language Models as Probabilistic Priors for Perception and Action
  ☆36Updated last year
• agopal42 / permakey
  Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…
  ☆9Updated 2 months ago
• siddharthverma314 / clcp-neurips-2020
  Code for Continual Learning of Control Primitives
  ☆18Updated 4 years ago
• Snagnar / Hieros
  Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper
  ☆15Updated 8 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
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆50Updated 3 months ago
• YunzhuLi / VGPL
  [ICML 2020] Visual Grounding of Learned Physical Models
  ☆39Updated 4 years ago
• wildphoton / Compositional-Generalization
  [NeurIPS 2022] Compositional Generalization in Unsupervised Compositional Representation Learning: A Study on Disentanglement and Emergen…
  ☆13Updated 2 years ago
• yuqingd / cusp
  ☆15Updated 2 years ago
• RobertCsordas / linear_layer_as_attention
  The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …
  ☆16Updated last year
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 2 years ago
• fabienbaradel / cophy
  "CoPhy: Counterfactual Learning of Physical Dynamics", F. Baradel, N. Neverova, J. Mille, G. Mori, C. Wolf, ICLR'2020
  ☆33Updated 4 years ago
• anirudh9119 / zforcing_nips17
  Z Forcing: Training Stochastic RNN's, NIPS'17
  ☆32Updated 7 years ago
• radarFudan / Curse-of-memory
  Curse-of-memory phenomenon of RNNs in sequence modelling
  ☆19Updated this week
• xingchenwan / bgpbt
  [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)
  ☆28Updated 2 years ago
• shwinshaker / LipGrow
  An adaptive training algorithm for residual network
  ☆15Updated 4 years ago
• davidbau / sidn-handbook
  The Structure and Interpretation of Deep Networks Handbook
  ☆12Updated 3 months ago