Alternatives and similar repositories for jax-vs-pytorch

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

• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆29Updated 4 years ago
• filteredcophy / FilteredCoPhy
  ☆10Updated 3 years ago
• bdusell / stack-attention
  Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"
  ☆18Updated last year
• rujiewu / Bongard-OpenWorld
  This is the official code implementation of Bongard-OpenWorld (ICLR 2024).
  ☆14Updated last year
• ToruOwO / mimex
  PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"
  ☆16Updated 2 years ago
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆32Updated 3 years ago
• yuqingd / cusp
  ☆15Updated 3 years 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 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 4 years ago
• davidbau / sidn-handbook
  The Structure and Interpretation of Deep Networks Handbook
  ☆14Updated last year
• subho406 / agalite
  AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)
  ☆22Updated last year
• allenai / signal-and-noise
  Measuring the Signal to Noise Ratio in Language Model Evaluation
  ☆28Updated 5 months ago
• Shalev-Lifshitz / MultiAgentVerification
  Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers
  ☆26Updated 10 months ago
• shwinshaker / LipGrow
  An adaptive training algorithm for residual network
  ☆17Updated 5 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• yilundu / ebm_compositionality
  [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models
  ☆47Updated 2 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 3 years ago
• facebookresearch / RRL
  Code for RRL (https://sites.google.com/view/abstractions4rl)
  ☆27Updated 4 years ago
• Snagnar / Hieros
  Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper
  ☆20Updated last year
• NingMiao / InstaAug
  ☆14Updated 3 years ago
• facebookresearch / ssorl
  Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories
  ☆42Updated 2 years ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆57Updated last year
• smonsays / hypernetwork-attention
  Official code for the paper "Attention as a Hypernetwork"
  ☆47Updated last year
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆38Updated last year
• facebookresearch / gwil
  Cross-Domain Imitation Learning via Optimal Transport
  ☆25Updated 3 years ago
• haoliuhl / tux
  Tools and Utils for Experiments (TUX)
  ☆15Updated 11 months ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆45Updated last year
• kylehkhsu / tripod
  ☆12Updated last year
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 3 years ago
• shoaibahmed / metadata_archaeology
  Official code for the paper: "Metadata Archaeology"
  ☆19Updated 2 years ago