jax

da76ebf0 - Removed inspect.Signature hackery from ``pl.BlockSpec``

Removed inspect.Signature hackery from ``pl.BlockSpec`` I realized it is unnecessary and is no different than listing the parameters in __init__ with relaxed types (to allow old argument order). PiperOrigin-RevId: 648696510