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
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆27Updated 3 years ago
• yuqingd / cusp
  ☆15Updated 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…
  ☆12Updated 3 months ago
• yilundu / ebm_compositionality
  [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models
  ☆45Updated 2 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 2 years ago
• bdusell / stack-attention
  Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"
  ☆17Updated last year
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆32Updated 2 years ago
• ruotianluo / lazy_related_work
  ☆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 2 weeks ago
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated 11 months ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 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 7 months ago
• agopal42 / permakey
  Code repository complementing the ICLR 2021 paper "Unsupervised Object Keypoint Learning using Local Spatial Predictability" (https://arx…
  ☆9Updated 5 months ago
• facebookresearch / dualformer
  implementation of dualformer
  ☆17Updated 3 months ago
• weigq / openview_quicklook
  ☆34Updated 3 months ago
• smonsays / hypernetwork-attention
  Official code for the paper "Attention as a Hypernetwork"
  ☆39Updated last year
• xingchenwan / bgpbt
  [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)
  ☆28Updated 2 years ago
• luckystarufo / SIREN_out_of_sample_tests
  A quick test of SIREN on out-of-sample tasks
  ☆13Updated 4 years ago
• minyoungg / overparam
  ☆41Updated 2 years ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆51Updated 6 months ago
• rosieyzh / openrlhf-pretrain
  Code for "Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretraining"
  ☆18Updated 2 months ago
• maxreciprocate / offline
  Offline RL experiments
  ☆15Updated 2 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆14Updated last year
• Shalev-Lifshitz / MultiAgentVerification
  Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers
  ☆19Updated 3 months ago
• Snagnar / Hieros
  Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper
  ☆16Updated 11 months ago
• belindal / LaMPP
  Code for LaMPP: Language Models as Probabilistic Priors for Perception and Action
  ☆37Updated 2 years ago
• cassidylaidlaw / orpo
  ☆16Updated 7 months ago