This job posting is currently inactive. Check out other active jobs at NodeRisk.

AI/ML Engineer

NodeRisk
Full-timeColombo, Sri Lanka

We’re an early-stage, venture-backed cybersecurity startup building AI- Powered Penetration Testing Platform . We’re looking for a founding AI/ML Engineer to architect and develop agentic systems powered by LLMs. You’ll design reasoning pipelinesand enable our system to execute real-world tasks autonomously.

What You’ll Do

  • Design and build LLM-powered agents capable of reasoning and executing multi-step workflows.
  • Architect and optimize prompt pipelines, memory systems, and decision graphs to improve task reliability and success rates.
  • Develop scalable pipelines for fine-tuning, evaluating, and deploying custom LLMs.
  • Integrate external tools, APIs, and graph-based data (TypeDB/Neo4j) to enhance context and reasoning.
  • Build Python frameworks to orchestrate tool execution, asynchronous tasks, and result collection.
  • Collaborate with full-stack engineers to expose AI outputs through APIs and dashboards.
  • Implement robust testing, monitoring, and logging for model-based workflows.
  • Research and prototype cutting-edge approaches to improve model reasoning, planning, and tool use.
  • Participate in architectural decisions and product planning as a core member of the founding engineering team.

What We’re Looking For

  • 3+ years of experience building ML/AI systems in production (startups or high-growth companies a plus).
  • Strong Python programming skills with experience in async frameworks and orchestrating workflows.
  • Hands-on experience with OpenAI, HuggingFace Transformers, or similar LLM APIs.
  • Familiarity with LangChain, AutoGen, Semantic Kernel, or other agent orchestration frameworks.
  • Knowledge of vector stores, embeddings, and semantic search.
  • Ability to integrate graph data (TypeDB, Neo4j) into AI pipelines.
  • Comfort working in a fast-moving, ambiguous environment with high autonomy.
  • Strong communication skills for collaborating with cross-functional teams.

Why Join Us

  • Founding AI/ML role with significant influence on architecture, product, and culture.
  • Meaningful equity in a venture-backed company at the cutting edge of AI + security.
  • Work with state-of-the-art models and agent frameworks to solve real-world challenges.
  • Remote-first and flexible work culture
  • High-impact environment where your work directly shapes a next-generation AI system.
  • Access to a learning budget, top-tier hardware, and room to experiment with emerging tools.

Apply for this job

Resume/CV*

Click or drag file to this area to upload your Resume

Please make sure to upload a PDF

First Name*
Last Name*
Email*
Phone Number*
The hiring team may use this number to contact you about this job.

By clicking 'Submit Application', you agree to receive job application updates from NodeRisk via text and/or WhatsApp. Message frequency may vary. Reply STOP to unsubscribe at any time. Message & data rates may apply.