During May 2025, Zhuo enhanced the tensorflow/tensorflow repository by adding support for U32 and U64 data types to the SPMD partitioner’s gather index functionality. This C++ feature expanded the partitioner’s ability to handle 32- and 64-bit indices directly, reducing the need for type casting and streamlining data pipelines that rely on varied index sizes. Zhuo’s work focused on robust data structure design and parallel computing principles, ensuring long-term maintainability and extensibility for future type support. The implementation addressed a specific gap in data representation flexibility, laying a solid foundation for more adaptable indexing paths in distributed workloads.