Differentiable and scalable 3D aerodynamics, (optionally) on the GPU.

Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust

Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrenderer/wiki; PR: https://github.com/google/brax/pull/367

Black Hole -> Three.js -> TSL -> (WebGPU/WebGL)

A simple JAX-powered simulation library for numerical experiments of fuzzy dark matter, stars, gas + more!

Efficient Differentiable n-d PDE solvers in JAX.

A JAX-based adaptive mesh refinement framework

N-dimensional adaptive mesh refinment tree structure in Python

GoMLX: An Accelerated Machine Learning Framework For Go

Code for the paper Universal Physics Transformers

webgpu autograd library

Geometry Processing Library for Python

Generate Streamlit apps from Tesseracts, automagically.

Selects tests affected by changed files. Executes the right tests first. Continuous test runner when used with pytest-watch.

PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation

Simons Stellarator Optimizer Code

the open-source turbomachinery designer

A Streamlit app for generating 3D model of the UAV wing console

Blade Builder 3D - 3-Dimensional Blade Surface CAD Generator

SU2: An Open-Source Suite for Multiphysics Simulation and Design

A compact, high-performance finite element analysis engine built on JAX.

A Python interface for the Dawn WebGPU engine

Real-time Simulation of Industrial Particulate Flows

You like pytorch? You like micrograd? You love tinygrad! ❤️

A retro game engine for Python

A community-maintained Python framework for creating mathematical animations.

fast distance and ray intersection queries

Universal, autodiff-native software components for Simulation Intelligence. 📦