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 last year
• normal-computing / thermox
  Exact OU processes with JAX
  ☆50Updated 4 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
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• vandijklab / Intelligence_at_the_edge_of_chaos
  ☆55Updated 5 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆209Updated 2 months ago
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆44Updated 9 months ago
• SamsungSAILMontreal / AnyMolGenCritic
  ☆21Updated 4 months ago
• misko / human_descent
  ☆38Updated 7 months ago
• DavidRuhe / clifford-group-equivariant-neural-networks
  Code used by the "Clifford Group Equivariant Neural Networks" paper.
  ☆84Updated last year
• adamnemecek / traceoid.ai
  ☆96Updated 8 months ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated 2 years ago
• profConradi / ALife
  Artificial Life simulations
  ☆52Updated 4 months ago
• microsoft / folx
  Implementation of Forward Laplacian algorithm in JAX
  ☆79Updated 2 months ago
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆30Updated 2 years ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 7 months ago
• Frost-group / LeadCopperApatite
  Modelling the new Lead-Copper apatite proposed room temperature supeconductor
  ☆32Updated 2 years ago
• EelcoHoogendoorn / numga
  Geometric algebra in JAX and numpy
  ☆82Updated 4 months ago
• thomasahle / tensorgrad
  Machine Learning with Symbolic Tensors
  ☆324Updated 2 months ago
• Artur-Galstyan / statedict2pytree
  ☆43Updated 2 months ago
• google-deepmind / torax
  TORAX: Tokamak transport simulation in JAX
  ☆456Updated this week
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated last year
• microsoft / cliffordlayers
  ☆159Updated last year
• abao1999 / panda
  Patched Attention for Nonlinear Dynamics
  ☆156Updated this week
• ml-jku / GINNs-Geometry-informed-Neural-Networks
  ☆172Updated last month
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆127Updated last month
• google-deepmind / flows_for_atomic_solids
  ☆52Updated 2 years ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆255Updated last month
• proximafusion / vmecpp
  From-scratch C++ and Python reimplementation of the Variational Moments Equilibrium Code (VMEC).
  ☆189Updated last week