AI Engineer

We're seeking exceptional AI Engineers to join our rapidly growing team and tackle some of the most complex challenges at the intersection of artificial intelligence and energy systems. You'll work on mission-critical problems including battery state prediction, energy demand forecasting, thermal management optimization, and autonomous energy trading systems that operate at massive scale.

This role offers the opportunity to work with world-class engineers, deploy AI systems that impact millions of users, and directly contribute to solving climate change through technology. You'll be building production systems that must operate with extreme reliability while pushing the boundaries of what's possible with AI.

As an AI Engineer, here’s what you’ll be doing

• 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

• 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 across diverse market conditions
• 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, particularly in domains relevant to energy systems, physics-informed neural networks, 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 that incorporate physical constraints and energy system dynamics into neural networks

• Work closely with hardware engineers to optimize AI algorithms for edge deployment
• Partner with product teams to define AI-powered features that deliver exceptional user experiences
• Collaborate with safety and regulatory teams to ensure AI systems meet automotive and grid-scale safety standards
• Support business development by demonstrating AI capabilities to potential customers and partners

• Understanding of battery technologies, electrochemistry, or power systems engineering
• Experience with physics-informed machine learning or scientific computing
• Background in control systems, signal processing, or embedded systems development
• Knowledge of energy markets, grid operations, or renewable energy systems
• Experience with automotive or aerospace safety-critical system development
• Familiarity with open-source AI ecosystems and model fine-tuning techniques (LoRA, QLoRA, PEFT)
• Experience with prompt engineering, few-shot learning, and in-context learning for foundation models

• Track record of shipping ML products that operate reliably at scale
• Strong software engineering fundamentals with experience in system design and architecture
• Comfort working in fast-paced environments with rapidly evolving requirements
• Experience with A/B testing, model evaluation, and performance monitoring in production
• Ability to work effectively across multiple codebases and collaborate with diverse engineering teams
• Hands-on experience with model serving infrastructure (vLLM, TGI, Ollama) and API development
• Proficiency with modern AI development workflows including model versioning, experiment tracking, and automated evaluation pipelines

• First principles thinking and ability to break down complex problems into manageable components
• Data-driven decision making with strong intuition for when and how to apply different ML techniques
• Comfortable with ambiguity and able to drive projects forward with minimal supervision
• Strong communication skills and ability to explain complex technical concepts to diverse audiences

Do you want to join a fast-growing climate tech startup with real impact?

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.

Coulomb is headquartered in San Francisco with its development and research center in Bangalore. Founders Khushboo Shrivastava and Santanu Mondal are alumni of IIT Bombay and have first-hand experience working on first-generation Electric Vehicles at General Motors. We are backed by some of the best climate tech investors in the world, including Y Combinator, Harvard Management Seed Capital, CSVE Ventures, and prominent Silicon Valley-based Angels.

We already have a global presence with customers in 5 countries across UK, US, APAC and India

Yet, we’re still at the start of our journey, and we want exceptional people to join us!