Alternatives and similar repositories for Cosmos:

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

• timduignan / Scalable-Electrolyte-Simulation
  This repository contains the data and scripts necessary to reproduce the results presented in the paper: **"Scalable molecular simulation…
  ☆41Updated 7 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆58Updated 5 months ago
• misko / human_descent
  ☆36Updated last month
• normal-computing / thermox
  Exact OU processes with JAX
  ☆38Updated 3 months ago
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆32Updated last year
• profConradi / ALife
  Artificial Life simulations
  ☆51Updated 7 months ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated 2 weeks ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆40Updated 2 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆113Updated 2 weeks ago
• StijnWoestenborghs / gradi-mojo
  ☆37Updated last year
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆34Updated last year
• heidelberg-hepml / lorentz-gatr
  Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)
  ☆17Updated last week
• google-deepmind / flows_for_atomic_solids
  ☆48Updated 2 years ago
• arghyadutta / resources
  Links to books, webpages, essays, apps, and other stuff from the internet.
  ☆24Updated 3 months ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• google-research / e3x
  E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
  ☆96Updated 2 months ago
• EelcoHoogendoorn / numga
  Geometric algebra in JAX and numpy
  ☆74Updated 4 months ago
• cgarciae / simple_flow_matching
  ☆14Updated last month
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆70Updated 4 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated 3 months ago
• DavidRuhe / clifford-group-equivariant-neural-networks
  Code used by the "Clifford Group Equivariant Neural Networks" paper.
  ☆79Updated 7 months ago
• philipturner / molecular-renderer
  Renderer for molecular nanotechnology
  ☆66Updated 4 months ago
• cgarciae / jax-gravity
  ☆24Updated last year
• annalena-k / tutorial-dingo-introduction
  Tutorial for the DINGO package to analyze gravitational wave data using NPE
  ☆38Updated 6 months ago
• smsharma / eqnn-jax
  Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.
  ☆23Updated last month
• y1xiaoc / fwdlap
  Forward mode laplacian implemented in JAX tracer
  ☆29Updated 2 months ago
• microsoft / folx
  Implementation of Forward Laplacian algorithm in JAX
  ☆57Updated this week
• ziatdinovmax / NeuroBayes
  Fully and Partially Bayesian Neural Nets
  ☆45Updated last week
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆33Updated 6 months ago