Neural Magic is an early-stage AI software company democratizing high performance for deep learning models. Our goal is to reduce the cost and increase the performance of end-users deploying deep learning applications. Based on decades of research at MIT, Neural Magic has developed a software platform that allows developers to sparsify deep learning models to minimize footprint and run on CPUs at GPU speeds. Please look through ourwebsite andGitHub repos to get a feel of what we are about.

Founded by an award-winning team of computer scientists and researchers out of MIT, we are a venture-backed company headquartered in Davis Square, Somerville, MA. Our investors include Amdocs, Andreessen Horowitz, Comcast Ventures, NEA, and Pillar VC.

We are seeking a machine learning engineer to work closely with our product and research teams to develop SOTA deep learning software. This person will work closely 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!

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

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

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 remote jobs

Machine Learning Engineer

Machine Learning Research Scientist

HPC Engineer - GPU