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.
  ☆59Updated 6 months ago
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆40Updated 3 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…
  ☆41Updated 8 months ago
• normal-computing / thermox
  Exact OU processes with JAX
  ☆42Updated 4 months ago
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆29Updated last year
• cgarciae / simple_flow_matching
  ☆15Updated 2 months ago
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated last year
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated last month
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆12Updated 4 months ago
• Frost-group / LeadCopperApatite
  Modelling the new Lead-Copper apatite proposed room temperature supeconductor
  ☆31Updated last year
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• uzulim / hades
  Fast singularity detection with kernel
  ☆33Updated last year
• smsharma / eqnn-jax
  Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.
  ☆30Updated 2 weeks ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 8 months ago
• misko / human_descent
  ☆36Updated 2 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆45Updated 2 weeks ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆73Updated last week
• DavidRuhe / clifford-group-equivariant-neural-networks
  Code used by the "Clifford Group Equivariant Neural Networks" paper.
  ☆79Updated 8 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025).
  ☆81Updated 3 months ago
• maxxxzdn / jax-geometric
  Implementation of various equivariant models in JAX
  ☆12Updated 10 months ago
• ml-jku / GINNs-Geometry-informed-Neural-Networks
  ☆151Updated this week
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆114Updated 2 weeks ago
• microsoft / cliffordlayers
  ☆149Updated last year
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 3 years ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆37Updated last year
• adamnemecek / traceoid.ai
  ☆66Updated 2 months ago
• JonathanGorard / Categorica
  Categorica: a pure and applied category theory framework for the Wolfram Language
  ☆30Updated 10 months ago
• gerkone / segnn-jax
  Steerable E(3) GNN in jax
  ☆24Updated last year