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 an ML Engineer, you will work closely with our product and research teams to develop SOTA deep learning software. You will collaborate with our technical and research teams to develop training and deployment pipelines, implement model compression algorithms, and productize deep learning research. If you are someone who wants to contribute to solving challenging technical problems at the forefront of deep learning, this is the role for you!

Join us in shaping the future of AI!

Responsibilities

Use your understanding of machine learning to tackle meaningful technical problems
Collaborate with research and product development teams to build machine learning products
Prototype and implement appropriate ML algorithms, tools, and pipelines
Create and manage training and deployment pipelines
Collaborate with a cross-functional team about market requirements and best practices
Keep abreast of developments in the field

Requirements

Proven experience as a machine learning engineer or similar role
Solid knowledge of machine learning and deep learning fundamentals with experience in one or more of computer vision, NLP, speech, reinforcement learning, generative models, etc
Knowledge of common ML frameworks (like PyTorch or Keras) and libraries (like NumPy and scikit-learn)
Strong programming skills with proven experience implementing Python-based machine learning solutions
Experience with engineering and supporting ML pipelines in a popular ML framework such as PyTorch, TensorFlow, jax, etc
Experience with engineering and maintaining training and/or deployment pipelines for Generative models / NLG / LLMs
Ability to interpret and implement research ideas and algorithms
Creative, collaborative, and innovation-focused
Strong sense of project ownership and personal responsibility
Bachelor's in Computer Science, Mathematics or similar field

Benefits

Health Care Plan (Medical, Dental & Vision)
Retirement Plan (401k, IRA)
Paid Time Off (Vacation, Sick & Public Holidays)
Family Leave (Maternity, Paternity)
Short Term & Long Term Disability
Training & Development
Work From Home
Free Food & Snacks
Wellness Resources
Stock Option Plan

We are an equal opportunity employer. All applicants will be considered for employment without attention to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.

Neural Magic AI Jobs

Latest Jobs

Machine Learning Engineer