jax.experimental.pallas.mosaic_gpu.GMEM

Contents

jax.experimental.pallas.mosaic_gpu.GMEM#

jax.experimental.pallas.mosaic_gpu.GMEM = MemorySpace.GMEM#: Alias of jax.experimental.pallas.mosaic_gpu.MemorySpace.GMEM.