Machine Learning Engineer

Neural Magic is an early-stage AI software company that is democratizing high performance for deep learning models. Our aim is to lower costs and improve performance for end-users implementing deep learning applications. Rooted in years of research at MIT, Neural Magic has developed a software platform that enables developers to sparsify deep learning models to minimize footprint and operate on CPUs at GPU speeds. Please browse our website and GitHub repos to get a sense of what we do. Founded by an acclaimed team of computer scientists and researchers from MIT, we are a venture-backed company located in Davis Square, Somerville, MA. Our investors include Amdocs, Andreessen Horowitz, Comcast Ventures, NEA, and Pillar VC. We're looking for a machine learning engineer to work closely with our product and research teams to develop state-of-the-art deep learning software. This person will work closely with our technical and research teams to create training and deployment pipelines, implement model compression algorithms, and commercialize deep learning research. If you're someone who wants to contribute to solving challenging technical problems at the forefront of deep learning, this is the role for you! Responsibilities: 1. Apply your understanding of machine learning to tackle meaningful technical problems. 2. Collaborate with research and product development teams to build machine learning products. 3. Prototype and implement suitable ML algorithms, tools, and pipelines. 4. Create and manage training and deployment pipelines. 5. Collaborate with a cross-functional team on market requirements and best practices. 6. Stay up-to-date with developments in the field. Requirements: 1. Proven experience as a machine learning engineer or similar role. 2. Solid knowledge of machine learning and deep learning fundamentals with experience in one or more of the following areas: computer vision, NLP, speech, reinforcement learning, generative models, etc. 3. Knowledge of common ML frameworks (like PyTorch or Keras) and libraries (like NumPy and scikit-learn). 4. Strong programming skills with proven experience implementing Python-based machine learning solutions. 5. Experience with engineering and supporting ML pipelines in a popular ML framework such as PyTorch, TensorFlow, jax, etc. 6. Experience with engineering and maintaining training and/or deployment pipelines for Generative models / NLG / LLMs. 7. Ability to interpret and implement research ideas and algorithms. 8. Creative, collaborative, and focused on innovation. 9. Strong sense of project ownership and personal responsibility. 10. Bachelor's degree in Computer Science, Mathematics or a related field. Benefits: 1. Health Care Plan (Medical, Dental & Vision). 2. Retirement Plan (401k, IRA). 3. Paid Time Off (Vacation, Sick & Public Holidays). 4. Family Leave (Maternity, Paternity). 5. Short Term & Long Term Disability. 6. Training & Development. 7. Remote Work. 8. Free Food & Snacks. 9. Wellness Resources. 10. Stock Option Plan. We are an equal opportunity employer. All applicants will be considered for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.