At a glance
- Location
- Bangalore · India
About the role
- The AI Engineer will work across the full stack, training and evaluating models while deploying them into real-world systems.
- This role involves partnering with platform, product, and firmware teams to own the development and production lifecycle of AI models.
- The engineer will address challenges related to perception and understanding using visual, audio, and real-world signals.
Responsibilities
- Research and develop machine learning models for perception and understanding problems.
- Explore the integration of specialized and general-purpose models in production systems.
- Design data, training, and evaluation approaches for real-world conditions.
- Study model behavior, robustness, and failure modes across deployment conditions.
- Integrate and validate new capabilities in real-time or resource-constrained systems.
- Collaborate with firmware, platform, and product teams to transition research into working systems.
Requirements
- Understanding of deep-learning fundamentals including architectures, training dynamics, and evaluation design.
- Ability to think in systems regarding latency, memory budgets, and on-device constraints.
- Experience taking end-to-end ownership across model, edge, and cloud boundaries.
- Proficiency in Python and PyTorch.
- Working ability in at least one systems language such as C++ or Go.
- Hands-on experience using agentic development tools and AI-assisted coding.
Skills
- Python
- PyTorch
- C++
- Go
- SNPE
- TensorRT
- TFLite
- ONNX Runtime
- Ray
- Anyscale
- Snowflake
- Redash
