ML Engineer/Scientist, AI Platform

About this job

Nightfall AI is an AI-native data loss prevention and insider risk management platform that protects sensitive information across SaaS applications, email, endpoint devices, and generative AI tools for numerous customers like top banks and AI innovators. The company's agentic platform automates remediation and helps security teams prevent data exfiltration at scale. As an ML Engineer or Scientist for the AI Platform team, you will help shape the future of data security by working alongside skilled engineers to design, build, and enhance machine learning models that drive the company's security products. A normal day in this role involves writing code in Python and Go, experimenting with natural language processing and modern language model architectures, fine-tuning models, benchmarking accuracy, building streaming microservices, and collaborating closely with product and engineering teams to push live AI features.

Responsibilities

• Build and ship AI systems that power Nightfall security products.
• Use Python for data analysis, model experimentation, fine-tuning, benchmarking accuracy, and real-time inferencing.
• Use Go to develop scalable microservices using distributed caching and streaming infrastructure.
• Collaborate with engineering and product teams to turn concepts into active AI features.
• Participate in the team on-call rotation to maintain operational excellence.

Requirements

• 5+ years of industry experience.
• Graduate degree in Computer Science, Engineering, AI, Machine Learning, Data Science, or a related field.
• Fluent coding skills in Python and Go, C++, or Java with strong knowledge of algorithms and data structures.
• Practical experience building deep learning models and deploying them to production at a large scale.
• Experience building agentic workflows.
• Ability to thrive in a fast-paced, high-ownership setting.

Good to have

• Hands-on experience with PyTorch or other frameworks for NLP, LLM fine-tuning, or agentic workflows.
• Familiarity with cloud platforms like AWS and GCP, alongside Kafka, Redis, and Postgres.
• Experience with MLOps across training, inference, evaluation, monitoring, and deployment.
• Open-source contributions, research papers, or hackathon wins.

Who should apply

This role suits experienced machine learning professionals with a strong background in artificial intelligence and software engineering who want to build impactful security products at scale. Candidates should have at least five years of industry experience and a relevant graduate degree. If you enjoy experimenting with advanced language models, writing clean code in Python and Go, and working in a high-ownership, fast-moving hybrid environment in Bengaluru, this position is a great fit for your career growth.

Frequently asked questions