AI Engineer
About the role
AI Engineer role at Coulomb AI, a climate tech startup building battery observability and AI-driven battery management solutions. The role focuses on designing, deploying, and maintaining production-grade ML models for battery state estimation, energy forecasting, and autonomous energy systems, with strong emphasis on edge deployment, observability, and collaboration with hardware and product teams.
Responsibilities
- Design, implement, and deploy machine learning models for battery state estimation, degradation prediction, and performance optimization
- Develop advanced time-series forecasting models for energy demand, supply, and pricing
- Design and validate ML models for anomaly detection and failure prediction
- Create reinforcement learning algorithms for optimal charging strategies, grid balancing, and energy arbitrage
- Build robust data pipelines processing terabytes of sensor data from batteries, inverters, and grid connections
- Develop monitoring and observability systems to ensure AI models maintain performance in production
- Stay at the forefront of AI research relevant to energy systems, physics-informed ML, and optimization
- Collaborate with battery scientists and power electronics engineers to translate domain expertise into AI solutions
- Publish research and represent Coulomb AI at top-tier conferences (NeurIPS, ICML, ICLR, IEEE Power & Energy)
- Build novel architectures incorporating physical constraints and energy system dynamics into neural networks
- Work with hardware engineers to optimize AI for edge deployment; collaborate with product, safety, and regulatory teams
- Support business development by demonstrating AI capabilities to customers and partners
Qualifications
- MS/PhD in Computer Science, Electrical Engineering, Physics, or related field, or equivalent practical experience
- 3+ years of experience developing and deploying machine learning systems in production environments
- Expert-level proficiency in Python, PyTorch/TensorFlow, and modern MLOps tools (MLflow, Kubeflow, etc.)
- Strong background in time-series analysis, optimization algorithms, or computer vision
- Experience with distributed computing frameworks (Ray, Dask, Spark) and cloud platforms (AWS, GCP, Azure)
- Proficiency with transformer architectures and modern NLP libraries (Hugging Face Transformers, LangChain, etc.)
- Experience building and deploying Retrieval-Augmented Generation (RAG) systems for technical documentation and knowledge management
- Hands-on experience fine-tuning and deploying open-source foundation models (DeepSeek, Llama, Mistral, etc.)
- Familiarity with vector databases (Pinecone, Weaviate, Chroma) and embedding models for semantic search applications
Skills
About Coulomb AI
Battery Observability Platform for Electric Vehicles
At Coulomb AI, we envision a future powered by the seamless integration of AI and battery technologies, creating a powerful synergy to combat climate change. We build category-leading software that can significantly improve battery lifespan and performance by employing advanced predictive analytics, real-time monitoring, and adaptive battery management algorithms.
More open roles at Coulomb AI
- Senior Full Stack Developer10mo agoHybrid (Bangalore) · Engineering
- Senior Data Scientist10mo agoHybrid (Bengaluru, KA, IN) · Engineering, Data science
- Senior Frontend Developer - Analytics Platform10mo agoHybrid (Bengaluru) · Engineering
- Senior Backend Engineer11mo agoHybrid (Bangalore)
- UX Designer11mo agoHybrid (Bengaluru, India)