jax.extend.xla.PipelineStage# class jax.extend.xla.PipelineStage(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)[source]# __init__(*args, **kwds)# Attributes PRE_SCHEDULER POST_SCHEDULER
