Research Engineer - AI Systems

Remote Remote · United States

pythonawspytorch engineeringfulltime

ashby

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

Join Yotta Labs as a Research Engineer focusing on AI systems optimization. Work remotely to enhance the performance of AI applications on a multi-silicon cloud platform.

Details

Remote work available globally
Full-time position
Skills required: Python, AWS, PyTorch

The work

Design and implement high-performance kernels for AI applications
Optimize kernels for NVIDIA, AMD, and AWS Trainium
Develop custom operators and graph optimizations
Design scalable distributed training and inference solutions