Robuta

https://docs.jax.dev/en/latest/jax.experimental.pallas.mosaic_gpu.html jax.experimental.pallas.mosaic_gpu module — JAX documentation jax experimental mosaic gpu https://docs.jax.dev/en/latest/_autosummary/jax.experimental.sparse.BCOO.html jax.experimental.sparse.BCOO — JAX documentation jax experimental sparse bcoo https://docs.jax.dev/en/latest/jax.experimental.serialize_executable.html jax.experimental.serialize_executable module — JAX documentation jax experimental serialize https://docs.jax.dev/en/latest/_autosummary/jax.experimental.sparse.bcoo_fromdense.html jax.experimental.sparse.bcoo_fromdense — JAX documentation jax experimental sparse bcoo