Alternatives and similar repositories for galax

Users that are interested in galax are comparing it to the libraries listed below

• google-deepmind / annealed_flow_transport

  View on GitHub
  ☆52Feb 27, 2023Updated 2 years ago
• cscarv / riemannian-metric-learning-ot

  View on GitHub
  ☆18Jun 26, 2023Updated 2 years ago
• Bellman-devs / bellman

  View on GitHub
  Model-based reinforcement learning in TensorFlow
  ☆56Jul 27, 2021Updated 4 years ago
• necludov / action-matching

  View on GitHub
  Implementation of Action Matching for the Schrödinger equation
  ☆25Jun 18, 2023Updated 2 years ago
• pairlab / vagram

  View on GitHub
  [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.
  ☆25Apr 15, 2023Updated 2 years ago
• srush / learns-dex

  View on GitHub
  ☆33Jan 14, 2021Updated 5 years ago
• bc / retina

  View on GitHub
  Tools for visualizing and comparing data from vertebrate retinas
  ☆14Jan 20, 2025Updated last year
• FlorianWilhelm / lda4rec

  View on GitHub
  🧮 Extended Latent Dirichlet Allocation for Collaborative Filtering in Recommender Systems.
  ☆42May 16, 2022Updated 3 years ago
• sparisi / td-reg

  View on GitHub
  TD-Regularized Actor-Critic Methods
  ☆36Dec 26, 2019Updated 6 years ago
• rolandgvc / flaxvision

  View on GitHub
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆45Jul 19, 2025Updated 6 months ago
• rlouf / mcx

  View on GitHub
  Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
  ☆329Mar 20, 2024Updated last year
• Information-Fusion-Lab-Umass / NuX

  View on GitHub
  Normalizing Flows using JAX
  ☆86Nov 30, 2023Updated 2 years ago
• cbg-ethz / LongSom

  View on GitHub
  LongSom tool for long-reads
  ☆11May 20, 2025Updated 8 months ago
• danieledurante / ESBM

  View on GitHub
  Extended stochastic block models with application to criminal networks
  ☆12Jan 16, 2022Updated 4 years ago
• Baran-phys / Tropical-Attention

  View on GitHub
  [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"
  ☆23Oct 23, 2025Updated 3 months ago
• KaiyangZ / PRIMUS

  View on GitHub
  Poisson scRNA Integration of Mixed Unknown Signals
  ☆11Oct 7, 2024Updated last year
• alchemistry / soft-benchmarks

  View on GitHub
  Soft benchmark sets for alchemical free energy calculations, for assessing their accuracy on biomolecular systems
  ☆16Mar 8, 2019Updated 6 years ago
• cbg-ethz / SGS

  View on GitHub
  Inference in Bayesian Networks with R
  ☆12Feb 6, 2023Updated 3 years ago
• j-towns / vdvae-jax

  View on GitHub
  Very deep VAEs in JAX/Flax
  ☆46Jun 16, 2021Updated 4 years ago
• hr0nix / dejax

  View on GitHub
  Accelerated replay buffers in JAX
  ☆46Sep 17, 2022Updated 3 years ago
• AmirTag / OT-ICNN

  View on GitHub
  Learning the optimal transport map via input convex neural neworks
  ☆42Jun 26, 2020Updated 5 years ago
• cvxgrp / cone_prog_refine

  View on GitHub
  Cone program refinement
  ☆10Mar 6, 2020Updated 5 years ago
• w4nderlust / cppn-tensorflow

  View on GitHub
  Very Simple and Basic Implementation of Compositional Pattern Producing Network in TensorFlow
  ☆11Nov 27, 2019Updated 6 years ago
• berndprach / AOL

  View on GitHub
  Code for paper Almost-Orthogonal Layers for Efficient General-Purpose Lipschitz Networks
  ☆12Aug 9, 2022Updated 3 years ago
• facebookresearch / w2ot

  View on GitHub
  Euclidean Wasserstein-2 optimal transportation
  ☆46Aug 19, 2023Updated 2 years ago
• webyrd / summer-scheming

  View on GitHub
  Summer Scheming!!!!!!
  ☆11Aug 20, 2020Updated 5 years ago
• optframe / pyoptframe

  View on GitHub
  Python bindings for OptFrame C++ Functional Core
  ☆12May 18, 2025Updated 8 months ago
• vEnhance / evans-phd-notebook

  View on GitHub
  Virtual notebook that Evan uses for his PhD thesis.
  ☆10Sep 5, 2025Updated 5 months ago
• google / jax-recommenders

  View on GitHub
  ☆11Oct 29, 2022Updated 3 years ago
• vwxyzjn / gym-pysc2

  View on GitHub
  Gym wrapper for pysc2
  ☆10Sep 16, 2022Updated 3 years ago
• jsiek / arete

  View on GitHub
  Arete is an experimental programming language.
  ☆12Oct 6, 2023Updated 2 years ago
• lgcarrier / xrppaperwallet

  View on GitHub
  ☆11Dec 31, 2024Updated last year
• Genentech / MMFM

  View on GitHub
  ☆19Nov 25, 2024Updated last year
• LaRiffle / approximate-models

  View on GitHub
  Python Library for Function Approximation in Machine Learning
  ☆12Nov 5, 2019Updated 6 years ago
• pucrs-ai-cs / reinforcement

  View on GitHub
  Reinforcement Learning
  ☆12Jun 22, 2017Updated 8 years ago
• MIT-REALM / efppo

  View on GitHub
  ☆11Mar 5, 2024Updated last year
• kilian-group / LMLM

  View on GitHub
  ☆23Oct 31, 2025Updated 3 months ago
• mschulth / rhc

  View on GitHub
  Implementation of Receding Horizon Curiosity Algrithm
  ☆13Mar 24, 2023Updated 2 years ago
• QubesOS / qubes-desktop-linux-i3-settings-qubes

  View on GitHub
  ☆12Dec 22, 2025Updated last month