[PJRT C API] Add a new PJRT extension (and related methods) for layouts.
[PJRT C API] Add a new PJRT extension (and related methods) for layouts.

The end goal is to provide new C APIs for layouts and remove the old ones. This is the first cl.
