The Staff AI/ML Engineer guides portfolio companies on AI product development, creates prototypes, experiments with AI tools, and advises on best practices.
Who We Are: Cox Exponential (CX2) is a unique take on a venture fund, started by a group of former ML researchers and technical cofounders who are passionate about the nuts and bolts of company building. We’re a cross between an early-stage investment firm, a technical accelerator, and an incubator. We don’t just write checks; we actively partner with founders to help grow their team and product vision.
If you love the energy of startups and the flexibility of academia, but also value financial stability and work/life balance, consider joining us at CX2!
We are looking for a creative, driven engineer with a background in AI and Machine Learning, who can provide technical guidance and support to our portfolio companies. In this role, you will help founders vet and steer the use of AI in their products, advise on best practices and common pitfalls, run experiments using a variety of methods and tools, and partner with your peers to our engineering organization to develop end-to-end prototypes of AI-powered features. While the focus will often be on modern language-based and agentic systems, you are encouraged to draw from a broader ML repertoire: The simplest solutions are often the best!
As Staff AI/ML Engineer, you will report directly to a Managing Partner at CX2, and be responsible for:
- Leveraging deep experience in AI and ML to help steer our portfolio company’s product roadmaps
- Experimenting with a variety of AI/ML tools to gauge their efficacy at solving specific, practical problems
- Partnering with a team of engineers (both within CX2 and our portfolio companies) to develop full, end-to-end prototypes of AI-powered features
- Acting as a sounding board for founders and their teams as they build AI products, including advising against common pitfalls
- Keeping up-to-date on the latest advances in GenAI, with a particular focus on Large Language Models, Agentic Systems, and related concepts (vector search, fine-tuning, prompt engineering, orchestration systems, etc)
- Aiding in the technical due diligence process, when appropriate, by vetting the feasibility and defensibility of prospective companies’ ML-powered products
- 5+ years of experience shipping AI/ML-powered products to real end users; the more variety, the better!
- 2+ years of specific experience building working products which leverage modern Generative AI techniques, with an emphasis on LLMs and agentic systems
- A rich understanding of the pragmatic aspects of working with AI in the modern era: prompt engineering, multi-agent verification systems, RAG, trade-offs in performance vs compute costs, etc.
- A background in more traditional Machine Learning techniques, including hands-on experience in training, evaluating, and deploying custom models for bespoke tasks
- Fluency in two or more programming languages, and comfort in learning new ones as the technical stack demands
- Excellent communication skills, and a willingness to work directly with clients
- Flexibility, enthusiasm, and the confidence to tackle challenges outside of your comfort zone
- Transparency; comfort with asking questions, voicing gaps in your understanding, and accepting when a particularly promising approach hits a dead end – they often do!
- A Bachelors or higher in Computer Science or another software engineering-related field
- A Masters or higher in an AI/ML adjacent field, or demonstrable experience in conducting open-ended AI/ML research in a non-academic environment
- Experience in a role which saw you juggling multiple research problems in parallel. (Examples: Technical consulting in an AI/ML related field; an Architect role at a large company; directing a team of AI/ML researchers in a corporate environment; mentoring multiple students)
- Experience in a high-paced startup environment
- Fluency in lower level languages and optimization (C/C++, OpenGL, Cuda)
- A passion for, and history of, open source development
- Demonstrated experience in communicating technical concepts to a nontechnical audience (teaching, blogging, public speaking, etc)
Top Skills
AI
C/C++
Cuda
Generative Ai
Large Language Models
Machine Learning
Opengl
Similar Jobs
Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Develop and deploy machine learning solutions for autonomous vehicles, mentoring engineers and collaborating on model implementations.
Top Skills:
Machine LearningPythonPyTorch
Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
As a Staff ML Infra Engineer, you will develop and deploy offboard machine learning solutions for autonomous vehicles, ensuring model integration and performance across teams. You'll build ML infrastructure, implement CI/CD pipelines, support data curation, and mentor engineers.
Top Skills:
Ci/CdDockerKubernetesNumpyPythonPyTorch
Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Lead the design, implementation, and deployment of scalable ML infrastructure for autonomous driving. Drive technical projects, mentor others, and collaborate with teams to enhance ML workflows.
Top Skills:
C++DockerKubernetesMlopsPythonPyTorchTensorFlow
What you need to know about the Los Angeles Tech Scene
Los Angeles is a global leader in entertainment, so it’s no surprise that many of the biggest players in streaming, digital media and game development call the city home. But the city boasts plenty of non-entertainment innovation as well, with tech companies spanning verticals like AI, fintech, e-commerce and biotech. With major universities like Caltech, UCLA, USC and the nearby UC Irvine, the city has a steady supply of top-flight tech and engineering talent — not counting the graduates flocking to Los Angeles from across the world to enjoy its beaches, culture and year-round temperate climate.
Key Facts About Los Angeles Tech
- Number of Tech Workers: 375,800; 5.5% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Snap, Netflix, SpaceX, Disney, Google
- Key Industries: Artificial intelligence, adtech, media, software, game development
- Funding Landscape: $11.6 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Strong Ventures, Fifth Wall, Upfront Ventures, Mucker Capital, Kittyhawk Ventures
- Research Centers and Universities: California Institute of Technology, UCLA, University of Southern California, UC Irvine, Pepperdine, California Institute for Immunology and Immunotherapy, Center for Quantum Science and Engineering
