Senior Software Developer AI/ML Engineer (Full Stack)

About the position

We're seeking a skilled Full Stack Software Developer to design and build generalized AI-driven solutions, such as Retrieval-Augmented Generation (RAG) systems and Agent orchestration frameworks to enable use of AI. You'll work across the entire technology stack using Rust and Python for backend development and modern frontend technologies to create intuitive user interfaces. This role combines expertise in AI application development with full stack engineering capabilities to deliver end-to-end solutions, from IDE plugins to evaluation pipelines for fine-tuned LLMs.

Responsibilities

• Design and develop full stack generalized AI solutions, including RAG systems and agent frameworks.
• Build high-performance, scalable backend applications using Rust for systems-level efficiency and Python for rapid prototyping and integration.
• Develop and maintain IDE plugins that enable agentic use of LLMs within development environments.
• Create intuitive frontend interfaces for AI applications using modern web technologies.
• Implement and optimize tool use functionalities for LLMs, enabling them to interact with external systems and APIs.
• Setup and maintain evaluation pipelines for fine-tuned LLMs using internal benchmarks and metrics.
• Integrate and optimize large language models (LLMs) and neural network architectures into production-ready software.
• Collaborate with cross-functional teams to define requirements and implement AI-driven features.
• Work with databases (e.g., SQL, NoSQL, vector databases) to manage and retrieve data efficiently for AI applications.
• Stay informed on the latest trends in AI, LLMs, neural networks, and related technologies to drive innovation.
• Write clean, maintainable, and well-documented code adhering to best practices.
• Troubleshoot and optimize AI applications for performance, reliability, and scalability.
• Contribute to the creation of reusable libraries and frameworks for AI development.

Requirements

• Experience building full stack AI applications, with a focus on practical, production-grade solutions.
• Strong proficiency in Rust for performance-critical systems and Python for AI development and scripting.
• Frontend development skills using modern JavaScript frameworks (React, Vue, or Angular).
• Experience developing IDE plugins or extensions (VS Code, JetBrains, etc.).
• Solid understanding of large language models (LLMs), their mechanics, and their real-world applications.
• Experience implementing tool use capabilities for LLMs and agent frameworks.
• Knowledge of evaluation methodologies for fine-tuned language models.
• Familiarity with neural network architectures and how they power modern AI systems.
• Experience with conventional databases and vector databases (e.g., PostgreSQL, MongoDB, OpenSearch, Qdrant).
• Good grasp of Retrieval-Augmented Generation (RAG) and latest AI Agent frameworks.
• Ability to stay current with the fast-evolving AI landscape, including advancements in LLMs and neural networks.
• Strong problem-solving skills and a creative approach to building generalized solutions.
• Bachelor's degree in Computer Science, Engineering, or a related field (or equivalent experience).

Nice-to-haves

• Experience with AI frameworks like TensorFlow, PyTorch, or Hugging Face Transformers.
• Familiarity with hardware-accelerated AI (e.g., GPUs, TPUs, or custom chips).
• Prior work with APIs or microservices for AI integration.
• Experience with CI/CD pipelines and DevOps practices.
• Knowledge of containerization technologies (Docker, Kubernetes).
• Experience with automated testing frameworks for both frontend and backend systems.

Benefits

• Competitive annual discretionary bonus program.
• Opportunity for annual RSU grants.
• Comprehensive benefits package designed to support success at work, at home, and at play.