Bo Zhang

Papers on Lattice

Total citations

Topics

h-index

Research focus

Architecture Design (Transformers, SSMs, MoE) (1)Multimodal Models (1)Robotics & Embodied AI (1)

Papers (1)

Jan 7, 2026

Bo ZhangJan 7, 2026

JAX-Shock: A Differentiable, GPU-Accelerated, Shock-Capturing Neural Solver for Compressible Flow Simulation

Differentiable fluid dynamics just got a speed boost: JAX-Shock offers a GPU-accelerated, high-order shock-capturing solver that lets you backprop through compressible flow simulations.

Bo Zhang

Architecture Design (Transformers, SSMs, MoE)Multimodal Models Robotics & Embodied AI

Search

Bo Zhang

Research focus

Papers (1)