JonathanGorard / CosmosLinks
The Cosmos numerical relativity code (with unstructured AMR)
☆20Updated last year
Alternatives and similar repositories for Cosmos
Users that are interested in Cosmos are comparing it to the libraries listed below
Sorting:
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated 11 months ago
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆44Updated 7 months ago
- Exact OU processes with JAX☆48Updated 3 months ago
- ☆43Updated 3 weeks ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- You should use PySR to find scaling laws. Here's an example.☆33Updated last year
- This repository contains the data and scripts necessary to reproduce the results presented in the paper: **"Scalable molecular simulation…☆47Updated last year
- Multiple dispatch over abstract array types in JAX.☆125Updated last week
- Stencil computations in JAX☆71Updated last year
- ☆53Updated 3 months ago
- Stochastic trace estimation using JAX☆15Updated 2 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 5 months ago
- Automatic Functional Differentiation in JAX☆72Updated 3 months ago
- Renderer for molecular nanotechnology☆67Updated last week
- Package of useful sampling algorithms written in MLX.☆17Updated last year
- ☆24Updated last year
- The code for "Diffusion Geometry" (2024).☆43Updated last year
- Neural Networks for JAX☆84Updated 9 months ago
- Transformers for Mathematics Tutorial | Simons/SLMath Workshop on AI for Mathematics 2025☆34Updated 2 months ago
- nanoGPT using Equinox☆13Updated 2 years ago
- ☆156Updated last year
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆26Updated 2 weeks ago
- This is a port of Mistral-7B model in JAX☆32Updated 11 months ago
- Categorica: a pure and applied category theory framework for the Wolfram Language☆31Updated last year
- Graph neural networks in JAX.☆67Updated last year
- A Wadler--Lindig pretty printer for Python☆39Updated this week
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 2 months ago
- ☆60Updated 3 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆83Updated last year
- ☆36Updated 6 months ago