Missing mask in JAX and Warp

Autodesk / XLB

XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML

Other

237 stars 25 forks source link

Closed hsalehipour closed 3 months ago

hsalehipour commented 3 months ago

Fixed the computation of missing mask for both JAX and Warp.
Major improvements to Warp impl of BC. Their signatures are all consistent now. This enabled using a single functinal (not 2d and 3d) for all BCs so far.