Machine Learning Engineer
Stackrox
Design and implement high-performance Python and C++ code for vLLM-based inference systems, GPU kernels, and numerical methods.
*Telecommuting permitted: work may be performed within normal commuting distance from the Red Hat, Inc. office in Boston, MA.
What You Will Do:
Develop, test, and optimize LLM inference algorithms, including quantization and sparsification techniques, to improve latency, throughput, and memory use.
Conduct performance profiling and modeling on NVIDIA GPUs using tools such as Nsight, tune CUDA, Triton, or CUTLASS kernels for deep neural networks.
Participate in technical design reviews and propose innovative HPC solutions for large-scale model serving.
Review peer code promptly and leverage AI-assisted development tools to uphold code quality standards.
Collaborate with cross-functional AI, product, and research teams to deliver features to Red Hat AI Inference Platform.
Document best practices and mentor engineers to foster a culture of continuous learning and open-source contribution.
Stay current with cutting-edge LLM optimization research (quantization, sparsity, speculative decoding) and integrate relevant advances into the product.
What You Will Bring:
Bachelor's degree (U.S. or foreign equivalent) in Computer Science, Computer Engineering or related field and two (2) years of experience in the job offered or related role.
Must have two (2) years of experience with: Python and Modern C++; CUDA, Triton, or CUTLASS kernel optimization; Deep learning frameworks, including PyTorch; performance analysis and profiling; and GPU memory architecture.
Must have one (1) year of experience with: LLM inference optimization; and distributed computing including inter and intra GPU communication protocols.
#LI-DNI
The salary range for this position is $133,650 - $220,680/year. Actual offer will be based on your qualifications.
Pay Transparency
Red Hat determines compensation based on several factors including but not limited to job location, experience, applicable skills and training, external market value, and internal pay equity. Annual salary is one component of Red Hat’s compensation package. This position may also be eligible for bonus, commission, and/or equity. For positions with Remote-US locations, the actual salary range for the position may differ based on location but will be commensurate with job duties and relevant work experience.
About Red Hat
Red Hat is the world’s leading provider of enterprise open source software solutions, using a community-powered approach to deliver high-performing Linux, cloud, container, and Kubernetes technologies. Spread across 40+ countries, our associates work flexibly across work environments, from in-office, to office-flex, to fully remote, depending on the requirements of their role. Red Hatters are encouraged to bring their best ideas, no matter their title or tenure. We're a leader in open source because of our open and inclusive environment. We hire creative, passionate people ready to contribute their ideas, help solve complex problems, and make an impact.
Inclusion at Red Hat
Red Hat’s culture is built on the open source principles of transparency, collaboration, and inclusion, where the best ideas can come from anywhere and anyone. When this is realized, it empowers people from different backgrounds, perspectives, and experiences to come together to share ideas, challenge the status quo, and drive innovation. Our aspiration is that everyone experiences this culture with equal opportunity and access, and that all voices are not only heard but also celebrated. We hope you will join our celebration, and we welcome and encourage applicants from all the beautiful dimensions that compose our global village.
Equal Opportunity Policy (EEO)
Red Hat is proud to be an equal opportunity workplace and an affirmative action employer. We review applications for employment without regard to their race, color, religion, sex, sexual orientation, gender identity, national origin, ancestry, citizenship, age, veteran status, genetic information, physical or mental disability, medical condition, marital status, or any other basis prohibited by law.