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.
  ☆61Updated 11 months ago
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆44Updated 7 months ago
• normal-computing / thermox
  Exact OU processes with JAX
  ☆48Updated 3 months ago
• Artur-Galstyan / statedict2pytree
  ☆43Updated 3 weeks ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated last year
• 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
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆125Updated last week
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• vandijklab / Intelligence_at_the_edge_of_chaos
  ☆53Updated 3 months ago
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆15Updated 2 months ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 5 months ago
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆72Updated 3 months ago
• philipturner / molecular-renderer
  Renderer for molecular nanotechnology
  ☆67Updated last week
• tedwards2412 / samplex
  Package of useful sampling algorithms written in MLX.
  ☆17Updated last year
• cgarciae / jax-gravity
  ☆24Updated last year
• Iolo-Jones / DiffusionGeometry
  The code for "Diffusion Geometry" (2024).
  ☆43Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 9 months ago
• wellecks / transformers4math-simons
  Transformers for Mathematics Tutorial | Simons/SLMath Workshop on AI for Mathematics 2025
  ☆34Updated 2 months ago
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆13Updated 2 years ago
• microsoft / cliffordlayers
  ☆156Updated last year
• heidelberg-hepml / lorentz-gatr
  Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)
  ☆26Updated 2 weeks ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 11 months ago
• JonathanGorard / Categorica
  Categorica: a pure and applied category theory framework for the Wolfram Language
  ☆31Updated last year
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• patrick-kidger / wadler_lindig
  A Wadler--Lindig pretty printer for Python
  ☆39Updated this week
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆95Updated 2 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• DavidRuhe / clifford-group-equivariant-neural-networks
  Code used by the "Clifford Group Equivariant Neural Networks" paper.
  ☆83Updated last year
• misko / human_descent
  ☆36Updated 6 months ago