Alternatives and similar repositories for Cosmos

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

• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆60Updated 10 months ago
• normal-computing / thermox
  Exact OU processes with JAX
  ☆48Updated 2 months ago
• misko / human_descent
  ☆36Updated 5 months ago
• heidelberg-hepml / lorentz-gatr
  Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)
  ☆25Updated this week
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆15Updated 2 months ago
• timduignan / Scalable-Electrolyte-Simulation
  This repository contains the data and scripts necessary to reproduce the results presented in the paper: **"Scalable molecular simulation…
  ☆47Updated last year
• flowersteam / sbmltoodejax
  ☆11Updated 5 months ago
• profConradi / ALife
  Artificial Life simulations
  ☆52Updated 2 months ago
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated last year
• JonathanGorard / Categorica
  Categorica: a pure and applied category theory framework for the Wolfram Language
  ☆31Updated last year
• annalena-k / tutorial-dingo-introduction
  Tutorial for the DINGO package to analyze gravitational wave data using NPE
  ☆42Updated 4 months ago
• MilesCranmer / pysr_paper
  A paper describing the implementation of PySR and SymbolicRegression.jl
  ☆58Updated last year
• uzulim / hades
  Fast singularity detection with kernel
  ☆33Updated last year
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆44Updated 6 months ago
• patrick-kidger / wadler_lindig
  A Wadler--Lindig pretty printer for Python
  ☆39Updated 3 weeks ago
• appliedgeometry / poissongeometry
  A Python module for (local) Poisson-Nijenhuis calculus on Poisson manifolds
  ☆13Updated 7 months ago
• cgarciae / jax-gravity
  ☆24Updated last year
• adamnemecek / traceoid.ai
  ☆95Updated 6 months ago
• vandijklab / Intelligence_at_the_edge_of_chaos
  ☆53Updated 3 months ago
• tedwards2412 / samplex
  Package of useful sampling algorithms written in MLX.
  ☆17Updated last year
• SamsungSAILMontreal / AnyMolGenCritic
  ☆21Updated 2 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆124Updated this week
• smsharma / eqnn-jax
  Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.
  ☆34Updated 3 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated 10 months ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• EelcoHoogendoorn / numga
  Geometric algebra in JAX and numpy
  ☆81Updated 2 months ago
• DavidRuhe / clifford-group-equivariant-neural-networks
  Code used by the "Clifford Group Equivariant Neural Networks" paper.
  ☆83Updated last year
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 5 months ago
• wellecks / transformers4math-simons
  Transformers for Mathematics Tutorial | Simons/SLMath Workshop on AI for Mathematics 2025
  ☆33Updated last month
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 11 months ago