Lightning AI is looking to solve the business problem of simplifying AI development and making it accessible to everyone by hiring a customer-facing Machine Learning Engineer.
Requirements
- Proficiency in ML frameworks such as PyTorch, TensorFlow, or JAX
- Familiarity with Lightning AI’s ecosystem is a plus
- Strong programming skills in Python
- Experience with common ML libraries and tools
- Proven ability to debug, optimize, and implement ML models and pipelines in production environments
- Experience with React and/or Go is a plus
- Knowledge of machine learning frameworks, tools, and best practices
Responsibilities
- Build LLM/CV Apps
- Write readable/testable/efficient code in Python, React, and/or Go
- Build and optimize ML models and pipelines for diverse use cases
- Design and deliver technical workshops, demos, and training sessions tailored to customer needs
- Troubleshoot and resolve complex technical challenges faced by customers
- Develop technical documentation, tutorials, and proof-of-concept solutions to empower customers and improve their ML workflows
- Stay up-to-date with the latest advancements in machine learning frameworks, tools, and best practices
Other
- 3–5+ years of hands-on experience with machine learning, data science, or a related field
- Customer-facing experience in a technical role (e.g., Solutions Engineer, ML Consultant) is highly desirable
- Exceptional written and verbal communication skills
- Experience working cross-functionally with engineering, product, and sales teams to deliver tailored solutions
- A degree in Computer Science, Data Science, or a related field, or equivalent practical experience
- Advanced degrees are a plus
- In-office requirements of 2 days per week in our New York City office
