CrawlJobs Logo
Balbix Logo Balbix · IT - Software Development

Staff AI Engineer

· Job Posted December 23, 2025
Apply Position
Job Link Share

Job Description

As a Staff AI Engineer you will get to play with petabyte data gathered from a multitude of data sources including Balbix proprietary sensors and 3rd party threat feeds. You will leverage a variety of AI techniques including deep learning, probabilistic graphical models, graph learning, recommendation systems, reinforcement learning, NLP, etc. And of course, you will be part of a team building a world-class product addressing one of the grand challenges in the technology industry.

Job Responsibility

  • Design and develop an ensemble of classical and deep learning algorithms for modeling complex interactions between people, software, infrastructure and policies in an enterprise environment
  • Design and implement algorithms for statistical modeling of enterprise cybersecurity risk
  • Apply data-mining, AI and graph analysis techniques to address a variety of problems including modeling, relevance and recommendation
  • Build production quality solutions that balance complexity and performance
  • Participate in the engineering life-cycle at Balbix, including designing high quality ML infrastructure and data pipelines, writing production code, conducting code reviews and working alongside our infrastructure and reliability teams
  • Drive the architecture and the usage of open source software library for numerical computation such as TensorFlow, PyTorch, and ScikitLearn

Requirements

  • Ph.D./M.S. in Computer Science or Electrical Engineering with hands-on software engineering experience
  • 5+ years of experience in the field of Machine Learning and programming in Python
  • Expertise in programming concepts and building large scale systems
  • Knowledge of state-of-the-art algorithms combined with expertise in statistical analysis and modeling
  • Robust understanding of NLP, Probabilistic Graphical Models, Deep Learning with graphs structures, model explainability, etc.
  • Foundational knowledge of probability, statistics and linear algebra

Nice to have

  • Able to take on very complex problems, learn quickly, iterate, and persevere towards a robust solution
  • Product-focused and passionate about building truly usable systems
  • Collaborative and comfortable working across teams including data engineering, front end, product management, and DevOps
  • Responsible and like to take ownership of challenging problems
  • A good communicator, and facilitate teamwork via good documentation practices
  • Comfortable with ambiguity and thrive in designing algorithms for evolving needs
  • Intuitive in using the right type of models to address different product needs
  • Curious about the world and your profession, constant learner
Balbix - All Job Offers

Looking for more opportunities?

Search for other job offers that match your skills and interests.

Suggested Tags: Frontend Developer Secretary Talent Manager Salesforce Administrator Data Analyst
Similar Jobs for

Staff AI Engineer

8 matching positions

Staff Ai Engineer

As a Staff AI Engineer, you will play a key role in designing and delivering hig...
Location
Location
India , Bengaluru
Salary
Salary:
Not provided
teradata.com Logo
Teradata
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's degree in Computer Science or equivalent from a recognized institution
  • 8+ years of experience in backend services, distributed systems, or data platform development
  • Strong proficiency in Java, Go, or Python for service development
  • Deep understanding of design principles, distributed system patterns, and service architecture
  • Hands-on experience designing and developing RESTful APIs
  • Experience with SQL and NoSQL databases and data modelling
  • Strong debugging, problem solving, and troubleshooting skills
  • Experience with modern containerization and orchestration tools such as Kubernetes
  • Knowledge of public cloud platforms
  • Experience with AI productivity tools (e.g., GitHub Copilot)
Job Responsibility
Job Responsibility
  • Design, architect, develop, and maintain high quality systems, services, and applications with an emphasis on scalability, reliability, and performance
  • Collaborate with cross-functional engineers and product partners to shape architecture and consistently deliver end to end features
  • Build and integrate robust RESTful APIs, ensuring security, data consistency, and maintainability
  • Work with SQL and NoSQL databases to implement efficient data models and service access patterns
  • Apply and experiment with AI/ML technologies, including agentic AI and large language models (LLMs)
  • Use AI powered engineering tools to improve development quality, speed, and productivity
  • Mentor engineers, supporting them in technical planning, implementation, and best practices
  • Identify and resolve system performance bottlenecks, optimizing code, architecture, and infrastructure
  • Write unit and integration tests and participate in code reviews to uphold engineering excellence
  • Investigate production issues, ensuring timely and effective solutions
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

Rapid7 is seeking a Staff AI Engineer to join our Data Science team as we expand...
Location
Location
India , Pune
Salary
Salary:
Not provided
rapid7.com Logo
Rapid7
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 8–12 years of experience as a Data Scientist, ML Engineer, or AI Engineer
  • Strong end-to-end practical expertise in ML/AI/DS
  • Able to explore, experiment, and deliver independently
  • Proficiency in scikit-learn, PyTorch, TensorFlow, Keras, HuggingFace, LangChain, Transformers, Pandas, NumPy
  • Strong communication skills
  • Collaboration across engineering, data science, product, and security teams
  • Experience mentoring and guiding junior data scientists
  • Agentic AI and LLM expertise
  • Hands-on experience with agentic AI architectures (CrewAI, LangGraph, LangChain agents, tool-calling, function-chaining)
  • Experience building multi-step reasoning workflows, autonomous agents, or LLM-powered decision systems
Job Responsibility
Job Responsibility
  • Work with security teams to define, scope, and design research efforts for new threat detections, automations, and AI-driven workflows
  • Collaborate with data scientists to transform research into production-ready solutions, mentoring on both methods and execution
  • Research, build, and evaluate ML and generative/LLM models, including agentic and multi-step reasoning systems
  • Design and optimise agentic architectures, including tool-calling, decision-making, memory, orchestration, and evaluation frameworks
  • Partner with engineering teams to ship AI features, integrating models into high-scale systems
  • Deploy AI/ML workloads in AWS using SageMaker, Bedrock, Lambda, EKS, Step Functions, and related services
  • Contribute to our LLMOps and MLOps workflows, improving evaluation pipelines, observability, reproducibility, and governance
  • Support development of AI security features, improving reasoning, context understanding, automation, and analyst workflows
  • Embrace agile development, iterative experimentation, and collaborative problem solving
  • Mentor junior team members and uplift the technical bar for agentic AI and ML engineering
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

As a Staff AI Engineer, you will be a senior technical leader responsible for de...
Location
Location
India , Hyderabad
Salary
Salary:
Not provided
teradata.com Logo
Teradata
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor’s degree in Computer Science, Engineering, or equivalent practical experience
  • 8+ years of experience building backend services, distributed systems, or data/AI platforms
  • Strong proficiency in Java, Go, or Python, with experience building large‑scale services
  • Deep understanding of distributed system design, scalability, fault tolerance, and cloud‑native architectures
  • Proven experience designing and operating production systems with SQL and NoSQL data stores
Job Responsibility
Job Responsibility
  • Lead the design and evolution of large‑scale, distributed AI systems that power Teradata’s AI platform and AI‑native products
  • Own end‑to‑end architecture for critical AI capabilities such as agentic workflows, RAG pipelines, vector search, semantic retrieval, and AI orchestration frameworks
  • Drive technical strategy and architectural consistency across multiple engineering teams
  • Design and implement production‑grade AI systems using LLMs, embeddings, vector databases, and agent‑based architectures
  • Build scalable, secure, and reusable platform services and APIs supporting AI workloads across the software development lifecycle
  • Define and implement guardrails for reliability, safety, governance, and cost control in enterprise AI systems
  • Partner with product management, architecture, research, and cloud platform teams to translate business requirements into scalable AI solutions
  • Influence roadmap decisions by providing deep technical insight, trade‑off analysis, and long‑term platform thinking
  • Act as a technical escalation point for complex system design, performance, and reliability challenges
  • Drive best practices for testing, observability, evaluation, and production readiness of AI systems
What we offer
What we offer
  • People-first culture
  • Flexible work model
  • Focus on well-being
  • Inclusive environment
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

We are looking for a visionary and technically deep AI Architect to lead the des...
Location
Location
India , Pune, Maharashtra; Karnataka
Salary
Salary:
Not provided
teradata.com Logo
Teradata
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 10+ years of experience in software engineering or technology architecture
  • At least 4 years focused on AI, ML, and Generative AI solutions at enterprise scale
  • Deep hands-on expertise with Large Language Models (LLMs) including model selection, prompt engineering, fine-tuning (LoRA, PEFT), and evaluation techniques
  • Proven experience designing and implementing MCP servers and structured context management strategies for LLM applications
  • Strong working knowledge of Generative AI frameworks and orchestration tools — LangChain, LlamaIndex, LangGraph, AutoGen, CrewAI, or equivalent agentic platforms
  • Hands-on experience with RAG architectures, vector databases (Pinecone, Weaviate etc), and embedding models
  • Proficiency in Python-based AI/ML development, including libraries such as HuggingFace Transformers, PyTorch, TensorFlow, scikit-learn, and OpenAI SDK
  • Experience with AI cloud platforms and services — Azure OpenAI, AWS Bedrock, Google Vertex AI, or equivalent enterprise AI infrastructure
  • Solid understanding of MLOps practices including model versioning, experiment tracking (MLflow, Weights & Biases), CI/CD for ML, and model monitoring in production
  • Familiarity with AI agent design patterns, multi-agent systems, tool-use frameworks, and autonomous workflow orchestration
Job Responsibility
Job Responsibility
  • Define and own the enterprise AI architecture strategy, establishing standards, reference architectures, and technology roadmaps for AI/ML adoption
  • Architect and oversee the design of LLM-powered applications including RAG pipelines, AI agents, and agentic workflow systems
  • Design and govern MCP server implementations, enabling structured, context-aware interactions between LLMs and enterprise data sources
  • Lead the evaluation and selection of AI/ML platforms, LLM providers (OpenAI, Anthropic, Azure OpenAI, Google Gemini, open-source models), and supporting infrastructure
  • Architect prompt engineering frameworks, fine-tuning pipelines, and model evaluation strategies to ensure LLM output quality, accuracy, and reliability
  • Design AI orchestration layers using frameworks such as LangChain, LlamaIndex, AutoGen, CrewAI, or custom agentic architectures for multi-step reasoning workflows
  • Establish AI data pipelines for model training, embeddings generation, vector database management, and real-time inference serving
  • Define and enforce responsible AI practices including bias detection, explainability, hallucination mitigation, content safety guardrails, and regulatory compliance
  • Collaborate with engineering leads, data scientists, and product teams to embed AI capabilities into existing platforms and new product initiatives
  • Mentor and upskill engineering teams on AI/ML technologies, architectural patterns, and emerging developments in the AI ecosystem
What we offer
What we offer
  • People-first culture
  • Flexible work model
  • Focus on well-being
  • Inclusive environment
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

The Staff Software Engineer will be responsible for building, maintaining, monit...
Location
Location
United States , Austin; San Francisco; San Mateo; Roseville; Irvine
Salary
Salary:
173000.00 - 200000.00 USD / Year
goodleap.com Logo
GoodLeap
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's degree
  • 7+ years experience in backend software, ML, or AI engineering
  • Highly proficient in ML and AI engineering tools, python/FastAPI, A2A protocol, agent design and evaluation, prompt and context engineering
  • Adept with agile software development lifecycle and DevOps principles
  • Passion for software development, emerging technologies and culture of innovation
  • Excellent communication and interpersonal skills
Job Responsibility
Job Responsibility
  • Collaborate with product, architecture, and design leads to deliver highly-available, fault-tolerant products and services
  • Work on significant and unique technical challenges, evaluate and recommend solutions, and guide decision making by considering technical tradeoffs
  • Grasp both the technical and business perspective so you can help drive innovation
  • Work autonomously and be self-disciplined, requiring no supervision or guidance
  • Mentor and coach the team members to grow both their technical skills and soft skills
What we offer
What we offer
  • May be eligible for a bonus and equity
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

As the staff engineer in our new Generative AI department, you will define the t...
Location
Location
Singapore , Singapore
Salary
Salary:
Not provided
airwallex.com Logo
Airwallex
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 10+ years in backend software development, with a passion for AI and language technologies
  • Proven track record of delivering major impact
  • Proficiency in Python, Typescript or Javascript
  • Experience in design and development of large-scale distributed, high concurrency, high load, high availability systems
  • Experience in vector databases
  • A relevant degree in Computer Science, Mathematics or related fields
Job Responsibility
Job Responsibility
  • Spearhead the creation of AI solutions to enable growth and drive internal efficiency across the business
  • Design and implement robust API and system architecture for new AI applications
  • Assist in AI application roadmapping and strategy
  • Collaborate effectively with a diverse, cross-functional team to leverage existing AI technologies, and research and implement new ones, to meet product goals
  • Providing prompt and effective solutions or workarounds in response to application issues or software bugs, maintaining high client satisfaction
  • Embrace the challenges of a fast-paced, high-growth environment with a focus on innovation
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

As we work towards building out the Context Layer for the Agentic Enterprise, we...
Location
Location
United States , Palo Alto
Salary
Salary:
Not provided
workato.com Logo
Workato
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's/Master's/PhD degree in Statistics, Mathematics, Computer Science, or another quantitative field
  • 7+ years of backend engineering experience with 3+ years in search, information retrieval, or related fields
  • Strong proficiency in Python
  • Hands-on experience with search engines (Opensearch or Elasticsearch)
  • Strong understanding of information retrieval concepts spanning traditional methods (TF-IDF, BM25) and modern neural search techniques (vector embeddings, transformer models)
  • Experience with text processing, NLP, and relevance tuning
  • Experience with relevance evaluation metrics (NDCG, MRR, MAP)
  • Experience with large-scale distributed systems
  • Proficiency in Knowledge Graph construction and optimization is a plus
  • Strong analytical and problem-solving skills
Job Responsibility
Job Responsibility
  • Lead the development of advanced query understanding systems that parse natural language, resolve ambiguity, and infer user intent
  • Design and deploy learning-to-rank models that optimize relevance using behavioral signals, embeddings, and structured feedback
  • Build and scale robust Entity Recognition pipelines that enhance document understanding, enable contextual disambiguation, and support entity-aware retrieval
  • Architect next-gen search infrastructure capable of supporting highly dynamic document corpora and real-time indexing
  • Create and maintain graph-based knowledge systems that enhance LLM capabilities through structured relationship data
  • Drive improvements in query rewriting, intent classification, and semantic search, using both statistical and neural methods
  • Own the design of evaluation frameworks for offline/online relevance testing, A/B experimentation, and continual model tuning
  • Collaborate with product and applied research teams to translate user needs into data-informed search innovations
  • Produce clean, scalable code and influence system architecture and roadmap across the relevance and platform stack
What we offer
What we offer
  • vibrant and dynamic work environment
  • multitude of benefits they can enjoy inside and outside of their work lives
Read More
Arrow Right

Staff AI Engineer

We're seeking a Staff AI Engineer to join our growing team and help build the ne...
Location
Location
Canada
Salary
Salary:
182000.00 - 220000.00 CAD / Year
procurify.com Logo
Procurify
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 10+ years of software engineering experience with a significant portion of that focussed on AI/ML, building and deploying AI features in production environments
  • Deep understanding of LLMs, prompt engineering, RAG architectures, vector databases (Databricks/pgVector), and agentic AI frameworks (LangChain, Google ADK, or similar)
  • Strong Python expertise with modern frameworks (FastAPI, Flask, Django) and microservice architecture patterns
  • Experience with AWS/GCP/Azure, containerization (Docker/Kubernetes), and CI/CD pipelines
  • Strong foundation in statistical methods, data pipelines, and ETL processes
  • Experience with A/B testing, evaluation frameworks, and hypothesis-driven development to validate AI features
  • Deep commitment to code quality, testing, monitoring, and observability in AI systems
  • Comfortable navigating ambiguity and rapidly adopting new AI technologies and methodologies
  • Strong communication skills with ability to explain technical concepts to both technical and non-technical stakeholders
Job Responsibility
Job Responsibility
  • Build AI-Powered Features: Design, develop, and deploy production-grade AI applications that solve real customer problems, from concept to deployment
  • Architect Scalable Systems: Create robust backend architectures that support AI workloads, ensuring low latency, high reliability, and seamless integration with existing services
  • Drive AI Innovation: Implement and optimize agentic AI systems, RAG pipelines, and multi-agent workflows using modern LLM frameworks
  • Own the Full Stack: Take end-to-end ownership of features, from database design and API development to AI model integration and monitoring
  • Collaborate Cross-Functionally: Work closely with product, design, and data teams to translate business requirements into technical solutions
  • Optimize and Iterate: Run experiments, A/B tests, and evaluations to continuously improve AI system performance and user experience
What we offer
What we offer
  • Flexible working: remote-first organization with flexible working hours
  • Responsible time off
  • Extended health benefits: competitive health, vision, and dental package along with an Employee Assistance Program (EAP), and a health and wellness spending account
  • Community initiatives: regular community days, Donate Your Day program, lunch and learns
  • Stock options
  • DEI initiatives: diversity, equity, and inclusion roundtable
  • Fulltime
Read More
Arrow Right