Research Engineer in Machine Learning

Neural Magic is an early-stage AI software firm that allows high-performance democracy for deep learning models. Our aim is to lower the cost and enhance the performance of end-users deploying deep learning apps. Based on many decades of research at MIT, Neural Magic has developed a software platform enabling developers to sparse deep learning models to reduce footprint and run on CPUs at GPU speeds. Please check out our website and GitHub repositories to understand what we do. We were founded by a recognized team of computer scientists and researchers from MIT, and we are a venture-backed company situated in Davis Square, Somerville, MA. Our investors include Amdocs, Andreessen Horowitz, Comcast Ventures, NEA, and Pillar VC. We are looking for a machine learning research engineer to closely work with our team on deep learning research and the use of model compression techniques. This individual will identify, report on, and create new algorithms and models within the deep learning domain. If you are looking to contribute to the solution of challenging technical issues at the forefront of deep learning, this role is for you! Responsibilities - Apply your understanding of machine learning to address meaningful technical difficulties. - Collaborate with research and product development teams to transform research ideas into product solutions. - Research and implement suitable ML algorithms and tools. - Conduct machine learning tests and experiments, optimizing hyperparameters. - Perform statistical analysis and fine-tuning using test outcomes. - Stay informed about developments in the field. Requirements - Verifiable experience as a machine learning researcher, engineer, or similar role. - Solid knowledge of machine learning and deep learning basics, along with experience in 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 and proven experience implementing Python-based machine learning solutions. - Experience with modeling algorithms and ML pipelines in popular ML frameworks such as PyTorch, TensorFlow, jax, etc. - Experience with Generative models/ NLG / LLMs training. - Ability to interpret and implement research ideas and algorithms. - Creativity, collaboration, and an innovation-focused approach. - Strong project ownership and personal responsibility sense. - Master's Degree in Computer Science, Mathematics, or a similar field. Benefits - Health Care Plan (Medical, Dental and Vision). - Retirement Plan (401k, IRA). - Paid Time Off (Vacation, Sick and Public Holidays). - Family Leave (Maternity, Paternity). - Short Term and Long Term Disability. - Training and Development. - Work From Home. - Free Food and Snacks. - Wellness Resources. - Stock Option Plan. We are an equal opportunity employer. All applicants will be considered for employment regardless of race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.