AI/Python Software Engineer (Onsite, Austin TX or Sunnyvale CA)

We are looking for an experienced AI Engineer with strong Python skills and hands-on experience in distributed computing using Ray. This role involves designing, developing, and deploying advanced AI solutions, building scalable ML pipelines, and optimizing workflows for high-performance environments. The ideal candidate combines deep knowledge of AI algorithms, distributed systems, and exceptional programming skills.

• Design and implement AI/ML models leveraging frameworks like TensorFlow, PyTorch, and scikit-learn.
• Develop robust Python code for data preprocessing, feature engineering, model training, and deployment.
• Build and maintain end-to-end ML pipelines, ensuring scalability and reliability.
• Leverage Ray for distributed training and parallel processing to optimize performance across large datasets.
• Collaborate with data scientists, engineers, and product teams to deliver AI-driven solutions.
• Deploy AI models on cloud platforms (AWS, Azure, GCP) and apply MLOps best practices.
• Document algorithms, workflows, and code for maintainability and compliance.

• Expert-level proficiency in Python and experience with AI/ML libraries (TensorFlow, PyTorch, scikit-learn, Pandas, NumPy).
• Strong understanding of machine learning algorithms, deep learning architectures, and data processing techniques.
• Hands-on experience with Ray for distributed computing and scaling ML workloads.
• Experience with cloud AI services (AWS SageMaker, Azure ML, GCP AI).
• Familiarity with CI/CD pipelines, containerization (Docker, Kubernetes), and Infrastructure-as-Code tools (Terraform).
• Knowledge of SQL and NoSQL databases for data handling.
• Strong problem-solving and debugging skills.

• Experience with Generative AI, LLMs, and prompt engineering.
• Knowledge of vector databases (FAISS, Pinecone) and RAG systems.
• Exposure to real-time AI systems, NLP, or computer vision.
• Contributions to open-source AI projects or published research.
• Cloud certifications (AWS, Azure, GCP) are a plus.

Experience: 5-8 Years
.

The expected compensation for this role ranges from $60,000 to $135,000 .

Final compensation will depend on various factors, including your geographical location, minimum wage obligations, skills, and relevant experience. Based on the position, the role is also eligible for Wipro's standard benefits including a full range of medical and dental benefits options, disability insurance, paid time off (inclusive of sick leave), other paid and unpaid leave options.

Applicants are advised that employment in some roles may be conditioned on successful completion of a post-offer drug screening, subject to applicable state law.

Wipro provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state, or local laws. Applications from veterans and people with disabilities are explicitly welcome.

Reinvent your world. We are building a modern Wipro. We are an end-to-end digital transformation partner with the boldest ambitions. To realize them, we need people inspired by reinvention. Of yourself, your career, and your skills. We want to see the constant evolution of our business and our industry. It has always been in our DNA - as the world around us changes, so do we. Join a business powered by purpose and a place that empowers you to design your own reinvention.