Autodesk / Autodesk/XLB

XLB in the Mosaic benchmark suite

Open
#165 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
507
Forks
85
PR merge metrics
No merged PRs in 30d

Description

We just released **Mosaic** ([github.com/pasteurlabs/mosaic](https://github.com/pasteurlabs/mosaic)), a benchmark for *differentiable* physics solvers. Each solver runs in a container behind a uniform `apply` / `vjp` interface, so one harness compares gradient accuracy and cost across JAX, PyTorch, Julia, and C++-adjoint codes on shared tasks.

**XLB is one of the benchmarked solvers** on the ns-grid and ns-3d-grid domains ([results](https://docs.pasteurlabs.ai/projects/mosaic/stable/docs/results.html) if you're interested).

We did our best to do it justice, but we don't know your code as well as you do and have surely made suboptimal calls. For example, we run BGK by default on the JAX backend (falling back to KBC for obstacles or low viscosity) and recover velocity from the lattice-Boltzmann populations the naive way. If a different collision operator such as MRT, a different relaxation time, or another macroscopic-conversion convention is the right choice for differentiable runs, we'd love to know.

The setup is self-contained under [`mosaic/tesseracts/navier-stokes-grid/xlb/`](https://github.com/pasteurlabs/mosaic/tree/main/mosaic/tesseracts/navier-stokes-grid/xlb/). If something looks odd, corrections or a PR are very welcome, here on GitHub or on the [Tesseract Forum](https://si-tesseract.discourse.group/).

Mostly we wanted you to know your work is represented, and to get it right.

Cheers,
Dion on behalf of the Mosaic authors

Contributor guide

Open the contributing guide

Research direction

Start by reviewing mosaic/tesseracts/navier-stokes-grid/xlb/ and its BGK/KBC, relaxation-time, and velocity-recovery choices. Compare those choices with XLB's behavior and identify concrete corrections or confirm the setup; the issue names no XLB files or tests to run.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Quiet
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.