AI/ML Architect

Balbix is the world’s leading platform for cybersecurity posture automation company. The Balbix Security Cloud uses AI and automation to reinvent how the World’s leading organizations reduce their cyber risk. With Balbix, security teams can accurately inventory their cloud and on-prem assests, conduct vulnerability management and quantify their cyber risk in monetary terms.

Balbix counts many global 1000 companies among its rapidly growing customer base . We are backed by John Chambers (the former CEO and Chairman of Cisco) , top Silicon Valley VCs and global investors . We have been called magical , and have received raving reviews as well as customer testimonials , numerous industry awards , and recognition by Gartner as a Cool Vendor , and by Frost & Sullivan .


At Balbix we believe in using the right algorithms and tools to ensure correctness, performance and deliver an excellent user experience. We draw boldly from the latest in AI/ML research but are unafraid to go beyond Bayesian inference and statistical models if the situation demands it. We are generalists, caring as much about storytelling with data, as about bleeding edge techniques, scalable model training and deployment.

We are building a data science culture with equal emphasis on knowing our data, grokking security-first principles, caring about customer needs, explaining our model predictions, deploying at scale, communicating our work, and adapting the latest advances.

We look out for each other, enjoy each others’ company, and keep an open channel of communication about all things data and non-data.

You Will:

  • 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

You Are:

  • 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

You Have:

  • A Ph.D. in Computer Science and Engineering or related field
  • 3+ years of experience in the field of Machine Learning, programming in Python, and building scalable distributed systems
  • Foundational knowledge of probability, statistics and linear algebra
  • Expertise in state-of-the-art AI algorithms such as deep-learning, NLP, probabilistic graphical models, graphical algorithms, reinforcement learning and time-series analysis
  • Knowledge of statistical analysis and modeling techniques, and model explainability

Life @ Balbix:

At Balbix, we have built a culture that aligns to our values of ownership, customer focus, curiosity, tenacity, innovation, judgement, teamwork, communication, honesty and impact. In joining our team you’ll work with very motivated and knowledgeable people, build pioneering products and utilize cutting-edge technology. Our Balbix team members see rapid career growth opportunities stemming from our culture of alignment, bottom up innovation, our clarity of goals and unrelenting mission. Last but not least, developing the world’s most advanced platform to address what the most important (and hardest) technology problem facing mankind today is exceptionally rewarding!

Benefits & Perks:

Balbix offers comprehensive medical, dental, vision, life insurance and long-term disability coverage for you and your family. Our Flex Time Off policy encourages you to take time off when you need it because we know and value how hard you work. When it comes to our offices it’s location, location, location we’re right next door to Santana Row so you can enjoy your time in (and out) of the office!

More information at

Please reach out if you want a seat on our rocket-ship and are passionate about changing the cybersecurity equation.

Job Types

Job Locations

Job Categories

Apply Now

Similar AI Jobs