AIML - ML Engineer -@Apple

🚀 Hiring: AI/ML Engineer @ Apple – Offshore Team (Remote, Full-Time, Hyderabad, Telangana)

Location:

Client:

Compensation:

Experience:

Minimum Qualifications

• Bachelors or M.S. in Computer Science (or related fields) or equivalent experience
• Expert in Python programming and tensor operations, including https://github.com/rougier/numpy-100
• Expert in at least one ML framework (PyTorch preferred but TensorFlow or JAX are also fine), including flash-attention, efficient kv-caching, DDP and FSDP
• Hands-on experience in establishing ML benchmarks (data, models, metrics)
• Involvement with open-source projects and experience with collaborative software development
• Excellent communication skills
• Readiness to encounter unforeseen challenges and to solve them

Preferred Qualifications

• Bachelors or M.S. in Computer Science (or related fields) or equivalent experience
• Expert in Python programming and tensor operations, including https://github.com/rougier/numpy-100
• Expert in at least one ML framework (PyTorch preferred but TensorFlow or JAX are also fine), including flash-attention, efficient kv-caching, DDP and FSDP
• Hands-on experience in establishing ML benchmarks (data, models, metrics)
• Active development in open-source projects and experience with collaborative software development
• Experience with CUDA programming, and/or High-Performance Computing and/or distributed computing
• Excellent communication skills
• Readiness to encounter unforeseen challenges and to solve them
• Past research experience
• C++ programming is a plus
• MLX experience is a plus

Description

Is this you? If so, we’d love to hear from you!

(Compensation range is indicative and will vary based on years of experience and skill fit.)