Key Responsibilities: - Implement latest algorithms from research papers for model compression in the optimization library. Apply these to the models critical for deployment and test on various architectures such as diffusion models, large language models etc. - Set up and debug training jobs, datasets, evaluation, performance benchmarking pipelines. Applying training time and post training compression techniques. Ability to ramp up quickly on new training code bases and run experiments. - Understanding HW capabilities and incorporating those in optimization algorithm design / enhancement.- Run detailed experiments and ablation studies to profile algorithms on various models, tasks, across different model sizes.- Improving model optimization documentation, writing tutorials and guides- Self prioritize and adjust to changing priorities and asks