Polymage Labs

A deep technology software startup building compiler technology for new AI chips and enabling acceleration and high performance for AI programming frameworks (like PyTorch, JAX) on existing hardware like GPUs through advanced program transformations. Incubated at the Indian Institute of Science (IISc), Bangalore. Founded by an IISc faculty member. Compilers are software that translate programs into instructions that the hardware can understand. The next generation of programming frameworks for AI hardware needs new compilers. PolyBlocks is PolyMage's compiler engine. It's modular, reusable, and retargetable, making it easy to seed highly effective compilers quickly for new hardware. PolyBlocks' advanced optimizations including cross-operator fusion and tiling exploit hardware performance maximally while unburdening programmers. Beats other state-of-the-art compilers and frameworks on existing popular hardware such as GPUs.