Alternatives and similar repositories for Tropical-Attention

Users that are interested in Tropical-Attention are comparing it to the libraries listed below

• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆63Updated 2 years ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆60Updated 7 months ago
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆38Updated last year
• facebookresearch / ssorl
  Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories
  ☆42Updated 2 years ago
• ml-jku / DIffUCO
  ☆63Updated 2 months ago
• google-deepmind / spectral_ssm
  ☆35Updated last year
• AndPotap / einsum-search
  ☆33Updated last year
• gengala / egnca
  Official implementation of E(n)-equivariant Graph Neural Cellular Automata
  ☆35Updated 2 months ago
• FLAIROx / cultural-accumulation
  ☆16Updated last year
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆41Updated last year
• Benjamin-Walker / selective-ssms-and-linear-cdes
  Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)
  ☆15Updated 11 months ago
• IDSIA / neuraldiffeq-fwp
  Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…
  ☆23Updated 6 months ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆29Updated last year
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆48Updated 2 years ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated last year
• facebookresearch / SSLForPDEs
  Code for Lie Symmetries SSL paper
  ☆24Updated last year
• neuroevobench / neuroevobench
  Neuroevolution Benchmark in JAX 🦕
  ☆42Updated 2 years ago
• Shalev-Lifshitz / MultiAgentVerification
  Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers
  ☆25Updated 9 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆56Updated 6 months ago
• Rose-STL-Lab / AutoSTPP
  Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…
  ☆25Updated last year
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆36Updated 4 years ago
• cambridge-mlg / neural_diffusion_processes
  ☆13Updated 2 years ago
• watcl-lab / positional_attention
  Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"
  ☆14Updated 7 months ago
• IdoAmos / not-from-scratch
  ☆33Updated last year
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆111Updated 2 years ago
• facebookresearch / LANCER
  Repo for the paper "Landscape Surrogate Learning Decision Losses for Mathematical Optimization Under Partial Information"
  ☆38Updated 2 years ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆39Updated last year
• mohmdelsayed / HesScale
  Scalable Computation of Hessian Diagonals
  ☆14Updated last year