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 last year
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆27Updated 3 years ago
• bdusell / stack-attention
  Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"
  ☆17Updated 11 months ago
• filteredcophy / FilteredCoPhy
  ☆10Updated 2 years ago
• facebookresearch / gwil
  Cross-Domain Imitation Learning via Optimal Transport
  ☆24Updated 2 years ago
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated 6 months ago
• yuqingd / cusp
  ☆15Updated 2 years ago
• shwinshaker / LipGrow
  An adaptive training algorithm for residual network
  ☆15Updated 4 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 2 years ago
• yilundu / ebm_compositionality
  [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models
  ☆44Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• libeanim / contact-symplectic-integrator-network
  Contact-Aware Symplectic Integrator Network
  ☆11Updated last year
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago
• siddharthverma314 / clcp-neurips-2020
  Code for Continual Learning of Control Primitives
  ☆18Updated 4 years ago
• PrincetonLIPS / MaM
  Official code for Generative Marginalization Models [ICML 2024] [SPGIM 2023 Workshop Oral]
  ☆22Updated 6 months ago
• StoneT2000 / trajectorytranslation
  Code for Abstract-to-Executable Trajectory Translation for One Shot Task Generalization (ICML 2023)
  ☆23Updated last year
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 3 months ago
• agopal42 / permakey
  Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…
  ☆9Updated last month
• nirgreshler / bayesian-online-planning
  The code for the paper "A Bayesian Approach to Online Planning" published in ICML 2024.
  ☆11Updated 8 months ago
• BruceGeLi / TCE_RL
  Temporally Correlated Episodic Reinforcement Learning, ICLR 24
  ☆12Updated 10 months ago
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆32Updated 2 years ago
• belindal / LaMPP
  Code for LaMPP: Language Models as Probabilistic Priors for Perception and Action
  ☆35Updated last year
• ling-pan / GAFN
  ☆23Updated 10 months ago
• luckystarufo / SIREN_out_of_sample_tests
  A quick test of SIREN on out-of-sample tasks
  ☆13Updated 4 years ago
• s-tian / mbold
  Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)
  ☆20Updated 3 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
• YunzhuLi / VGPL
  [ICML 2020] Visual Grounding of Learned Physical Models
  ☆40Updated 4 years ago
• allenai / robustnav
  Evaluating pre-trained navigation agents under corruptions
  ☆28Updated 3 years ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆30Updated 6 months ago
• bergen / EdgeTransformer
  ☆22Updated 3 years ago