Alternatives and similar repositories for jax-vs-pytorch

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

• filteredcophy / FilteredCoPhy
  ☆10Updated 2 years ago
• ssnl / poisson_quasimetric_embedding
  Open source code for paper "On the Learning and Learnability of Quasimetrics".
  ☆32Updated 2 years ago
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆27Updated 4 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 3 months ago
• shwinshaker / LipGrow
  An adaptive training algorithm for residual network
  ☆17Updated 5 years ago
• yilundu / ebm_compositionality
  [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models
  ☆46Updated 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• yuqingd / cusp
  ☆15Updated 3 years ago
• bdusell / stack-attention
  Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"
  ☆17Updated 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
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated last year
• nirgreshler / bayesian-online-planning
  The code for the paper "A Bayesian Approach to Online Planning" published in ICML 2024.
  ☆11Updated last year
• subho406 / agalite
  AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)
  ☆21Updated 11 months ago
• Shalev-Lifshitz / MultiAgentVerification
  Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers
  ☆22Updated 7 months ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 2 years ago
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 3 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆43Updated 11 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 3 years ago
• ruotianluo / lazy_related_work
  ☆15Updated 3 years ago
• allenai / signal-and-noise
  Measuring the Signal to Noise Ratio in Language Model Evaluation
  ☆23Updated last month
• e2crawfo / silot
  Original tensorflow implementation of SILOT (Spatially Invariant, Label-free Object Tracking).
  ☆13Updated 2 years ago
• davidbau / sidn-handbook
  The Structure and Interpretation of Deep Networks Handbook
  ☆14Updated 9 months ago
• belindal / LaMPP
  Code for LaMPP: Language Models as Probabilistic Priors for Perception and Action
  ☆37Updated 2 years ago
• rujiewu / Bongard-OpenWorld
  This is the official code implementation of Bongard-OpenWorld (ICLR 2024).
  ☆13Updated 9 months ago
• Yusufma03 / CVRL
  code for CoRL 2020 paper "Contrastive Variational Model-Based Reinforcement Learning for Complex Observations"
  ☆24Updated 3 years ago
• google-deepmind / affordances_option_models
  ☆23Updated 3 years ago
• aaronwalsman / ltron
  ☆64Updated last year
• vacancy / neural-logic-machines
  PyTorch implementation for the Neural Logic Machines (NLM).
  ☆11Updated 6 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆37Updated last year