Apply on Employer Site

Lightning AI · 1 month ago

Machine Learning Solutions Engineer

New York, New York, United States

Full-time

Hybrid

Mid Level

$120K/yr - $175K/yr

3+ years exp

Lightning AI is a company reimagining the way AI is built, aiming to simplify AI development for everyone. As a Machine Learning Solutions Engineer, you will serve as a technical expert, working with customers to optimize their machine learning workflows and collaborating with internal teams to enhance product features.

Computer VisionInformation TechnologyMachine LearningNatural Language ProcessingSoftware

H1B Sponsor Likely

Responsibilities

Write readable/testable/efficient code in Python

Partner with customers to understand their machine learning (ML) workflows, challenges, and goals, providing technical expertise to help them achieve success with Lightning AI’s platform

Scope and deliver proof-of-concepts that align customer goals with the strengths of our platform

Work with customers to build and optimize ML models and pipelines for diverse use cases, working hands-on with customer data and applications

Design and deliver technical workshops, demos, and training sessions tailored to customer needs

Collaborate closely with internal teams—including product, engineering, and sales—to relay customer feedback, advocate for feature enhancements, and influence the product roadmap

Develop technical documentation and tutorials to empower customers and improve their ML workflows

Troubleshoot and resolve complex technical challenges faced by customers, acting as their advocate within Lightning AI

Stay up-to-date with the latest advancements in ML frameworks, tools, and best practices, sharing knowledge with both internal teams and customers

Qualification

Machine LearningPythonML frameworksTechnical documentationCustomer-facing experienceCommunication skillsCross-functional collaborationProblem-solving

Required

3–5+ years of hands-on experience with machine learning, data science, or a related field

Understanding of machine learning concepts, tools, and best practices

Proficiency in Python and experience with ML frameworks such as PyTorch or TensorFlow

Proven ability to debug, optimize, and implement ML models and pipelines in production environments

Exceptional written and verbal communication skills, with the ability to explain technical concepts to both technical and non-technical audiences

Experience working cross-functionally with engineering, product, and sales teams to deliver tailored solutions

Thrive in a dynamic, customer-driven environment, balancing multiple projects and priorities simultaneously

A degree in Computer Science, Data Science, or a related field, or equivalent practical experience

Preferred

Customer-facing experience in a technical role (e.g., Solutions Engineer, ML Consultant) is highly desirable

Familiarity with Lightning AI's open source portfolio is a plus

Advanced degrees are a plus

Benefits

Medical, dental and vision

Life and AD&D insurance

Flexible paid time off plus 1 week of winter closure

Generous paid family leave benefits

$500 monthly meal reimbursement, including groceries & food delivery services

$500 one time home office stipend

$1,000 annual learning & development stipend

100% Citibike membership (NYC only)

$45/month gym membership

Additional various medical and mental health services

Company

Lightning AI

The AI development platform - From idea to AI, Lightning fast ⚡️. Code together. Prototype. Train on GPUs. Scale. Serve.

Founded in 2019

New York, New York, USA

11-50 employees

https://www.pytorchlightning.ai

H1B Sponsorship

Lightning AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2023 (2)

2021 (1)