ML Engineering Lead

Your Role in Our Mission:

As an AI Engineer Lead, you will play a pivotal role in leading and managing projects from concept to production, mentoring team members, and influencing the strategic direction of our AI initiatives. Knowledge of semiconductor design and manufacturing is a plus.

Responsibilities:

1. Lead AI projects from concept to production deployment

2. Solve challenging AI and software engineering problems while promoting best practices

3. Create showtime-ready benchmarks to continually measure quality and robustness of solutions relative to baselines

4. Develop and deploy state-of-the-art AI models for problems in hardware engineering with complex logical and uncertainty-bound constraints

5. Evaluate state-of-the-art Bayesian and non-Bayesian approaches to reliable deep learning and formal verification of AI systems

6. Set up experimentation tools and synthetic data infrastructure to support rapid experimentation and iteration, with a clear path to production deployment

7. Develop strategies to manage AI-specific challenges (latency, variance, errors)

8. Keep up with AI advancements, especially in language models and multi-modal AI, and synthetic data generation

What Makes You A Great Fit:

1. 4+ years of experience with deep learning frameworks like Pytorch, Tensorflow, Jax

2. Rich leadership experience over the “full stack” when it comes to designing, training, evaluating and deploying machine learning models, especially large generative models

3. Strong software engineering skills, especially in building complex, distributed systems around AI technologies

4. Expertise in prompt engineering, fine-tuning, and deploying large generative models in production environments

5. Skilled in handling and preprocessing large datasets for AI applications, including multimodal data

6. Strong understanding of AI evaluation metrics and benchmarking methodologies

7. Excellent communication skills, with the ability to explain complex AI concepts to technical and non-technical stakeholders

What Elevates Your Application:

1. Experience deploying AI models in high-stakes or regulated environments

2. Hands-on experience with cloud platforms for large-scale AI deployment

3. Familiarity with probabilistic programming languages (e.g., TensorFlow Probability, Pyro) and probabilistic reasoning methods (e.g. Bayesian NNs or Monte Carlo Tree Search)

4. Specialized knowledge in advanced AI techniques such as few-shot learning, meta-learning, or AI alignment, and relevant frameworks like DSPy

5. Contributions to open-source AI projects or publications in top-tier AI conferences/journals

6. Deep curiosity for or experience in semiconductors and physics

7. A "defensive AI engineering" mindset, with experience handling the challenges of working with non-deterministic AI systems