AI Trainee

About Rasta.AI

Rasta.AI is revolutionizing road safety and infrastructure monitoring using cutting-edge AI technologies. Our AI-powered solutions depend on high-quality data to predict, analyze, and improve road conditions globally.

Role Overview

We are looking for a passionate and motivated AI Intern to join our team. This role offers an opportunity to work on real-world AI problems and contribute to production-level solutions.

Key Responsibilities

• Design, develop, and implement machine learning and deep learning models for real-world applications.
• Work extensively on computer vision and natural language processing (NLP) tasks.
• Perform data collection, cleaning, preprocessing, and feature engineering.
• Analyze large datasets using NumPy and Pandas to extract meaningful insights.
• Build, test, and optimize AI/ML models using TensorFlow and PyTorch.
• Deploy and manage applications using Docker and work with cloud platforms like AWS or GCP.
• Work with databases (SQL/NoSQL) for data storage, retrieval, and management.
• Use version control tools like Git to maintain clean and collaborative codebases.
• Operate within Linux/Unix environments and leverage GPU acceleration (CUDA).

Requirements & Tech Stack

• Strong proficiency in Python programming.
• Solid understanding of Data Structures, Algorithms, and Software Engineering principles.
• Hands-on experience with Machine Learning and Deep Learning frameworks:
• TensorFlow
• PyTorch
• Practical knowledge of:
• Computer Vision (OpenCV)
• Natural Language Processing (NLP)
• Model optimization techniques
• Experience with data handling and analysis using:
• NumPy
• Pandas
• Data visualization libraries (Matplotlib, Seaborn, etc.)
• Familiarity with:
• Model deployment using REST APIs
• Containerization tools like Docker
• Cloud platforms such as AWS or GCP
• Knowledge of:
• Databases (SQL / NoSQL)
• Version control systems (Git)
• Linux/Unix environments
• GPU acceleration (CUDA)

Qualifications

• Currently pursuing a Bachelor’s degree in Artificial Intelligence, Machine Learning, Computer Science, or a related field.

Compensation Structure

• Stipend (Probation Period – 6 months): ₹10,000 per month
• Post Probation Period (Next 12 Months): ₹20,000 per month
• After 18 months: ₹3.4 – ₹5.4 LPA (Based on performance during 18 months)
• Note: Mandatory 18-month contract agreement.