Software Engineer

AO Labs (Headquarters: SF, CA + remote)

AI Machine Learning Reinforcement Learning Real-time WNN Weightless Neural Networks API Edge AI Software Engineering Backend API Development Python Distributed Systems MLOps AI Engineer Back End Engineer Data Engineer Staff Engineer

Company: AO Labs is an applied AI research lab building real-time reinforcement learning using continuously trainable, compute-efficient weightless neural networks (WNNs). We aim to unlock hyper-personalized AI with less data than traditional LLMs, creating models that learn continuously. Our first product is an API enabling fast, per-user learning loops by combining user context, application data, and feedback, allowing training at the edge without a gap between training and inference.

Role: We are hiring Software Engineers for various roles to build, optimize, and deploy our cutting-edge AI platform. Responsibilities include developing our core API product, implementing and scaling real-time learning systems using WNNs, designing infrastructure for edge training and inference, integrating with application data sources, and ensuring system reliability and performance. You will work closely with researchers to productionize novel AI/ML models and contribute to a system that bridges the gap between static pre-trained models and continuous, contextual learning.

Ideal Candidate: Strong software engineering background with proven experience building and scaling complex systems, particularly APIs and data-intensive applications. Proficiency in languages like Python, Go, or Rust, and familiarity with cloud platforms (AWS/GCP/Azure). Experience with MLOps, distributed systems, real-time data pipelines, or deploying ML models to edge devices is highly valuable. You should be passionate about applying engineering rigor to novel AI challenges and thrive in a collaborative, fast-paced startup environment.