AI Engineer

About Us

Astra is a cybersecurity SaaS company that makes otherwise chaotic pentests a breeze with its one of a kind AI-led offensive Pentest Platform. Astra's continuous vulnerability scanner emulates hacker behavior to scan applications for 13,000+ security tests.

CTOs and CISOs love Astra because it helps them to achieve continuous security at scale, fix vulnerabilities in record time, and seamlessly transition from DevOps to DevSecOps with Astra's powerful CI/CD integrations.

Astra is loved by 800+ companies across 70+ countries. In 2024 Astra uncovered 2.5 million+ vulnerabilities for its customers, saving customers $110M+ in potential losses due to security vulnerabilities.

We've been awarded by the President of France Mr. François Hollande at the La French Tech program and Prime Minister of India Shri Narendra Modi at the Global Conference on Cyber Security. Loom, MamaEarth, Muthoot Finance, Canara Robeco, Dream 11, OLX Autos etc. are a few of Astra’s customers.

Job Description

This is a remote position.

Role Overview

As Astra Security’s first AI Engineer, you will play a pivotal role in introducing and embedding AI into our security products. You will be responsible for designing, developing, and deploying AI applications leveraging both open-source models (Llama, Mistral, DeepSeek etc) and proprietary services (OpenAI, Anthropic). Your work will directly impact how AI is used to enhance threat detection, automate security processes, and improve intelligence gathering.

This is an opportunity to not only build future AI models but also define Astra Security’s AI strategy, laying the foundation for future AI-driven security solutions.

Key Responsibilities

• Lead the AI integration efforts within Astra Security, shaping the company’s AI roadmap
• Develop and Optimize Retrieval-Augmented Generation (RAG) Pipelines with multi-tenant capabilities
• Build and enhance RAG applications using LangChain, LangGraph, and vector databases (e.g. Milvus, Pinecone, pgvector).
• Implement efficient document chunking, retrieval, and ranking strategies.
• Optimize LLM interactions using embeddings, prompt engineering, and memory mechanisms.
• Work with Graph databases (Neo4j or similar) for structuring and querying knowledge bases
• esign multi-agent workflows using orchestration platforms like LangGraph or other emerging agent frameworks for AI-driven decision-making and reasoning.
• Integrate vector search, APIs and external knowledge sources into agent workflows.
• Exposure to end-to-end AI ecosystem like Huggingface to accelerate AI development (while initial work won’t involve extensive model training, the candidate should be ready for fine-tuning, domain adaptation, and LLM deployment when needed)
• Design and develop AI applications using LLMs (Llama, Mistral, OpenAI, Anthropic, etc.)
• Build APIs and microservices to integrate AI models into backend architectures..
• Collaborate with the product and engineering teams to integrate AI into Astra Security’s core offerings
• Stay up to date with the latest advancements in AI and security, ensuring Astra remains at the cutting edge
  
  

What We Are Looking For

• Exceptional Python skills for AI/ML development
• Hands-on experience with LLMs and AI frameworks (LangChain, Transformers, RAG-based applications)
• Strong understanding of retrieval-augmented generation (RAG) and knowledge graphs
• Experience with AI orchestration tools (LangChain, LangGraph)
• Familiarity with graph databases (Neo4j or similar)
• Experience in Ollama for efficient AI model deployment for production workloads is a plus
• Experience deploying AI models using Docker
• Hands-on experience with Ollama setup and loading DeepSeek/Llama.
• Strong problem-solving skills and a self-starter mindset—you will be building AI at Astra from the ground up.
  
  

Nice To Have

• Experience with AI deployment frameworks (e.g., BentoML, FastAPI, Flask, AWS)
• Background in cybersecurity or security-focused AI applications
  
  

What We Offer

Software Engineering Mindset:

This role requires a strong software engineering mindset to build AI solutions from 0 to 1 and scale them based on business needs. The candidate should be comfortable designing, developing, testing, and deploying production-ready AI systems while ensuring maintainability, performance, and scalability.

Why Join Astra Security?

• Own and drive the AI strategy at Astra Security from day one
• Fully remote, agile working environment.
• Good engineering culture with full ownership in design, development, release lifecycle.
• A wholesome opportunity where you get to build things from scratch, improve and ship code to production in hours, not weeks.
• Holistic understanding of SaaS and enterprise security business.
• Annual trips to beaches or mountains (last one was at Wayanad).
• Open and supportive culture.
• Health insurance & other benefits for you and your spouse. Maternity benefits included.