Machine Learning Engineer
Job description
A Machine Learning Engineer role at Loka building and improving machine learning and generative AI solutions across client projects, from LLM-powered applications to retrieval pipelines. The position is for an experienced ML engineer to ship GenAI into production, primarily in life sciences.
At a glance
- Work mode
- Remote
- Languages
- English – excellent
What you'll do
- Take ownership of assigned tasks and projects, accountability is key.
- Participate in client communications, communicating deliverables clearly.
- Understand business objectives and build solutions that help achieve them, along with metrics to track progress.
- Collaborate cross functionally with other teams (Data Engineering, DevOps, Backend).
- Build and maintain components of generative AI systems, including LLM powered application features and agentic workflows.
- Build information retrieval pipelines within established architectural patterns (data ingestion, chunking, embeddings, vector search).
- Contribute to evaluation frameworks and production monitoring for GenAI solutions.
- Wrangle, explore and visualize data with a keen eye for data cleaning.
- Deploy, maintain and upgrade ML models and pipelines, analyzing model errors and designing strategies to overcome them.
- Write and maintain technical documentation.
What we're looking for
- Bachelor's degree in Computer Science or a related field.
- 2+ years of AI/ML engineering experience.
- Hands-on experience building GenAI solutions, prompt engineering, fine-tuning and serving LLMs, search and embeddings, using common frameworks (LangChain/LangGraph, LlamaIndex).
- Understanding of statistical, ML and deep learning algorithms.
- Working knowledge of cloud ML services, preferably AWS (Bedrock, AgentCore, SageMaker).
- Proficiency in Python and core ML libraries and frameworks (scikit-learn, PyTorch, HuggingFace, TensorFlow, Transformers).
- Familiarity with containerization and orchestration tools.
- A collaborative mindset, with openness to feedback and eagerness to grow your craft.
- Willingness to support and learn from more senior teammates on the ML crew.
- Autonomy, adaptability and a consistently positive presence on the teams you work with.
- Excellent English, as a global team, we work entirely in English for meetings, customer calls and business communications.
- CV written in English.
Nice to have
- Building and deploying agents with common patterns (Agentic RAG, NLQ) and frameworks (smolagents, strands-agents).
- MLOps/LLMOps experience, preferably in AWS, along with standard tooling (MLFlow, LangFuse).
- Experience in consultancy environments or startups, including working directly with clients, managing projects and adapting to different settings.
Your team
ML crew
An experienced machine learning crew shipping GenAI into production across life sciences.
Benefits
- Every other Friday off (26 extra days off a year)
- Remote-first culture
- Explore and Relocation programs (three months work abroad or full international relo)
- Paid sick days and local holidays
- Business English classes program
- Continuous Learning Support
- Fitness and/or Mental Health Subscriptions
- Access to LokaLabs™, our internal research and development program
- Defined career path
Apply
About Loka, Inc
A globally distributed tech consultancy based in Silicon Valley.
Loka is a globally distributed tech consultancy based in Silicon Valley. We help clients ship machine learning and GenAI into production across life sciences, as well as marketing, advertising and more. We work entirely remotely, test new ideas in our in-house incubator LokaLabs™, observe local holidays and take every other Friday off.
Key facts
- more than 100 certified specialists, technical experts and PhDs
- Silicon Valley
- Tech consultancy
- Recognized by AWS as Innovation Partner of the Year in 2024, beating out 150,000 partners for the title.
- Counts more than 100 certified specialists, technical experts and PhDs among its teammates.
- Works entirely remotely.
- Runs an in-house incubator, LokaLabs™.
- Takes every other Friday off.
Partners & customers
What they value
- Curious
- Autonomous and positive
- Team player
- Adaptable
- Coachable
