Software Engineer (AI Products)

Team: Data Team

Location: Vilnius

Commitment: Full Time

Workplace Type: hybrid

The team waiting for you:

We’re forming a dedicated squad to take this from promising product to industry-leading platform. If you’re a Python developer who’s been itching to go deep on LLMs, RAG, and agentic AI — not just toy projects, but production systems at scale — this is your chance. You’ll work alongside experienced Staff Engineers, ship to real customers, and help define how AI reshapes B2B data intelligence. We have a seat ready for you!

In this role, you’ll:

• Design, develop, and maintain Python backend services powering our AI products.
• Build and optimize LLM-powered pipelines — RAG, semantic search, natural language query processing.
• Develop agent workflows and tool-use patterns for the AI Data Assistant.
• Integrate with large-scale data infrastructure — vector databases, search engines, and analytical query engines.
• Design and implement API endpoints to serve AI-generated insights to client-facing products.
• Optimize prompt engineering, LLM output quality, and system reliability.
• Collaborate with cross-functional teams (Data Engineering, API, Client Side) to deliver product features.
• Monitor and improve performance, latency, and cost efficiency of AI services.
• Contribute to evaluation frameworks and testing strategies for AI-powered features.
• Share knowledge and help build the team’s collective expertise in applied AI.

Your skills & experience:

• 2+ years of professional backend development experience.
• Experience building and deploying production-grade APIs.
• Familiarity with LLM APIs (OpenAI, Anthropic, or similar) — professional or personal projects.
• Understanding of retrieval-augmented generation (RAG) patterns or strong willingness to learn.
• Experience working with databases — relational (PostgreSQL/MySQL) and/or search engines (Elasticsearch).
• Experience with Docker and containerized environments.
• Strong problem-solving skills and genuine curiosity about AI technologies.
• Ability to work with ambiguity and evolving requirements in a fast-moving product space.

Nice to have:

• Experience with vector databases (Qdrant, Pinecone, Weaviate).
• Experience with LLM orchestration frameworks (LangChain, LlamaIndex).
• Experience with agent/tool-use patterns and agentic programming.
• Data engineering background — Kafka, Spark, Trino, dbt, or similar.
• Experience with prompt engineering and LLM evaluation.
• Experience with Kubernetes.
• Familiarity with observability tools (Grafana, Sentry, OTel).
• Understanding of NLP concepts (embeddings, semantic similarity, tokenization).

Salary & Benefits:

Gross salary: 3200-5600 EUR/month. Keep in mind that we are open to discuss a different salary based on your skills and competencies.
Growth & Learning: 40+ internal learning options, external conferences, mentorship, and year-round knowledge-sharing.
Health & Well-being: Private health insurance, psychotherapy, on-site well-being consultants, 24/7 gym access, and a wellness app.
Celebration & Community: Team events, an overseas workation, quarterly team-building budgets, and plenty of ways to mark milestones together.
Plus bonus vacation days, paid life-moment days off, barista coffee, and all the tools you need. See the full list of benefits here: https://career.oxylabs.io/blog/our-work-benefits/.