Software Engineer - vLLM (ML)

About Neural Magic

Based in Somerville, Massachusetts, Neural Magic is a series A startup backed by leading investors including Andreessen Horowitz, NEA, NEA, Pillar, VMware, Verizon Ventures, Comcast Ventures, and Amdocs. At Neural Magic we believe the future of AI is open and we are on a mission to bring the power of open-source LLMs and VLLM to every enterprise on the planet. Neural Magic accelerates AI for the enterprise and brings operational simplicity to GenAI deployments. As a leading developer and maintainer of the vLLM project and inventor of state-of-the-art techniques for model quantization and sparsification, Neural Magic provides a stable platform for enterprises to build, optimize and scale LLM deployments.

Our Mission

Neural Magic is on a mission to bring the power of open-source LLMs and vLLM to every enterprise on the planet.

Your Role

As a Software Engineer focused in vLLM, you will be at the forefront of innovation, collaborating with our team to tackle the most pressing challenges in model performance and efficiency. Your work with machine learning and high performance computing will directly impact the development of our cutting-edge software platform, helping to shape the future of AI deployment and utilization.

Join us in shaping the future of AI!

Responsibilities

• Write robust Python and C++, working on vLLM systems, high-performance machine learning primitives, performance analysis and modeling, and numerical methods.
• Review code and help develop best practices for the team.
• Work closely with machine learning teams to attain high performance for neural networks in the engine.

Requirements

• Extensive experience in writing high performance code for GPUs and deep knowledge of GPU hardware
• BS, MS, or PhD in computer science
• Mathematical software, especially linear algebra or signal processing
• Modern C++
• Python
• Pytorch
• Tensor computations
• Deep neural network models and techniques
• A self-directed practitioner who learns fast
• Strong communications skills with both technical and non-technical team members
• Strong sense of project ownership and personal responsibility
• Genuine interest in learning

Benefits

• Competitive compensation and stock option plan

• Comprehensive health care (medical, dental, vision)
• Retirement plan (401k, IRA)
• Generous paid time off (vacation, sick leave, holidays)
• Family leave (maternity, paternity)
• Disability coverage
• Professional development opportunities
• Flexible work arrangements (remote options)
• Wellness resources
• Free food and snacks (in the office)

Neural Magic is an equal-opportunity employer committed to fostering a diverse and inclusive workplace. All applicants will be considered for employment without attention to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.