jax.experimental.pallas.tpu.get_pipeline_schedule

jax.experimental.pallas.tpu.get_pipeline_schedule(schedule)

Retrieve a named pipeline schedule or pass through fully specified one.

Return type:

Any