Founding Machine Learning Engineer
Employment Information
Report this job
Job expired or something wrong with this job?
Job Description
Founding Machine Learning Engineer at A1 shaping an impactful AI consumer application. Overseeing model selection, training strategy, and infrastructure with full autonomy and influence.
Responsibilities:
- Build end-to-end training pipelines: data → training → eval → inference
- Design new model architectures or adapt open-source frontier models
- Fine-tune models using state-of-the-art methods (LoRA/QLoRA, SFT, DPO, distillation)
- Architect scalable inference systems using vLLM / TensorRT-LLM / DeepSpeed
- Build data systems for high-quality synthetic and real-world training data
- Develop alignment, safety, and guardrail strategies
- Design evaluation frameworks across performance, robustness, safety, and bias
- Own deployment: GPU optimization, latency reduction, scaling policies
- Shape early product direction, experiment with new use cases, and build AI-powered experiences from zero
- Explore frontier techniques: retrieval-augmented training, mixture-of-experts, distillation, multi-agent orchestration, multimodal models
Requirements:
- Strong background in deep learning and transformer architectures
- Hands-on experience training or fine-tuning large models (LLMs or vision models)
- Proficiency with PyTorch, JAX, or TensorFlow
- Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray)
- Strong software engineering skills — writing robust, production-grade systems
- Experience with GPU optimization: memory efficiency, quantization, mixed precision
- Comfortable owning ambiguous, zero-to-one technical problems end-to-end
Benefits:
- Extreme ownership and autonomy from day one - you define and build key model systems.
- Founding-level influence over technical direction, model architecture, and product strategy.
- Remote-first flexibility
- High-impact scope—your work becomes core infrastructure of a global consumer AI product.
- Competitive compensation and performance-based bonuses
- Backing of a profitable US$2B group, with the speed of a startup
- Insurance coverage, flexible time off, and global travel insurance
- Opportunity to shape a new global AI product from zero
- A small, senior, high-performance team where you collaborate directly with founders and influence every major decision.
Similar Jobs
Senior Manager – Machine Learning
Senior Manager leading the Machine Learning team to enhance CCC’s subrogation products. Collaborating with stakeholders and delivering innovative AI solutions in the insurance technology space.
Senior AI/ML Engineer
Senior AI/ML Engineer designing and implementing workflows leveraging LLMs at iBusiness. Collaborating with cross-functional teams to develop scalable AI-driven solutions.
Machine Learning Engineer – Mid-level
Machine Learning Engineer responsible for developing and operating scalable ML solutions at Extractta. Collaborating with product, data, and engineering teams to deliver robust models.
Senior Machine Learning Engineer
Senior Machine Learning Engineer at IQVIA leading ML applications design focusing on generative AI solutions. Collaborating with teams and guiding development of production-grade algorithms.
Senior Machine Learning Engineer
Senior Machine Learning Engineer developing core ML models and data pipelines for a gaming company. Seeking an autonomous engineer with deep passion for video games.
Machine Learning Engineer
Machine Learning Engineer developing AI safety solutions at Gray Swan. Focus on real-world impact in adversarial testing and model evaluation.
Staff Machine Learning Engineer – Generative AI, Full-Stack Applications
Senior Machine Learning Engineer developing AI solutions and full-stack applications for CVS Health. Collaborating with teams to design and prototype secure, resilient products and platform components.
Senior Machine Learning Engineer – Consultant
Senior AI/ML Engineer Consultant responsible for implementing AI solutions and mentoring juniors at leading AI consultancy Provectus. Join to solve complex technical challenges and impact enterprise clients.
Machine Learning Engineer
Machine Learning Engineer designing and evaluating deep learning models for agricultural robotics applications. Collaborating with engineers and product teams to enhance model performance in real-world scenarios.
Senior Machine Learning Engineer – AI Product
Machine Learning Engineer for AI Product team at Qonto, developing customer-facing AI solutions. Collaborating with stakeholders and ensuring integration of ML solutions into financial services.
ML Engineer – Agents, Data
Senior ML Engineer for AI product development in a leading social discovery company. Working on large-scale ML systems and cutting-edge generative AI technologies.
Senior MLOps
MLOps role enhancing ML infrastructure and deployment for a global travel-tech company. Focus on managing ML services and optimizing resources in a fully remote setup.
MLOps Engineer
ML Ops Engineer designing and maintaining ML workflows using AWS services. Collaborating with data scientists to ensure reliable and scalable models in production.
Machine Learning Lead
Machine Learning Lead driving AI initiatives for LexisNexis Legal & Professional, delivering scalable, robust solutions with hands-on leadership and mentoring.
Senior ML Engineer
Senior AI Engineer contributing to AI engineering teams at Relativity. Design and develop scalable AI systems, collaborating with cross-functional teams and driving engineering excellence.
AI/ML Engineer
AI/ML Engineer developing AI solutions for professional services firms. Working on production models and real data to enhance client relationships.
Machine Learning Engineer
Machine Learning Engineer developing AI capabilities within Zillow’s Agentic AI team. Focusing on production-grade systems that enhance real estate experiences through advanced reasoning.
Senior Machine Learning Engineer – ML/AI Team
Senior Machine Learning Engineer at A Place for Mom focused on developing ML and AI solutions. Designing and implementing advanced AI models to drive business impact using unstructured data.
Senior Machine Learning Engineer
Senior Machine Learning Engineer responsible for the full ML lifecycle at Button. Collaborating with data teams to build systems that integrate machine learning into commerce and monetization products.
Senior Machine Learning Engineer
Senior Machine Learning Engineer developing AI solutions for customer service at Replicant. Leading exploration of Large Language Models and Generative AI for innovative user experiences.