feat: JLL changes for IFRT Shardings (#770)

* feat: JLL changes for IFRT Shardings

* chore: run clang-format

* chore: run formatter

* fix: remove old APIs

* fix: prevent accidental mismatches between compile and execute

Author: avik-pal