Machine Learning Research Engineer at Etched.ai

The Machine Learning Research Engineer will propose and conduct novel research to achieve results on Sohu that are unviable on GPUs, translate core mathematical operations from the most popular Transformer-based models into maximally performant instruction sequences for Sohu, develop deep architectural knowledge informing best-in-the-world software performance on Sohu HW, collaborate with HW architects and designers, co-design and finetune emerging model architectures for highest efficiency on Sohu, guide and contribute to the Sohu software stack, performance characterization tools, and runtime abstractions by implementing frontier models using Python and Rust, propose and implement a novel test time compute algorithm that leverages Sohu’s unique capabilities, implement diffusion models on Sohu to achieve GPU-impossible latencies, optimize model instructions and scheduling algorithms, and implement model-specific inference-time acceleration techniques.