Founding ML Engineer
Job Description
Our R&D team is focused on creating the most effective engine for deploying generative AI models, with efforts ranging from precise GPU kernel fine-tuning to comprehensive system optimizations.
We're looking for an expert level engineer with a strong background in either CUDA, ROCm, or Triton kernel optimization. Your role will involve leading substantial improvements in GPU performance and playing a key role in pioneering AI and machine learning initiatives.
Job Responsibilities
- Explore and analyze performance bottlenecks in ML training and inference.
- Develop and optimize high-performance computing kernels in Triton, CUDA, and/or ROCm.
- Implement programming solutions in C/C++ and Python.
- Deep dive into GPU performance optimizations to maximize efficiency and speed.
- Collaborate with the team to extend and improve existing machine learning compilers or frameworks such as MLIR, Pytorch, Tensorflow, ONNX Runtime, TensorRT. (This is optional but beneficial)
Job Requirements
- Bachelor's, Master’s or PhD’s degree in Computer Science, Electrical Engineering, or a related field.
- Strong programming skills in C/C++ and Python.
- Deep understanding and experience in GPU performance optimizations.
- Proven experience with kernel optimizations on CUDA, ROCm, or other accelerators.
- General experience with the training and deployment of ML models
- Experience with distributed systems development or distributed ML workloads
- (Optional but preferred) Experience with innovative OSS projects like FlashAttention, mlc-llm, vllm.
- (Optional but preferred) Experience with machine learning compilers or frameworks such as TVM, MLIR, Pytorch, Tensorflow, ONNX Runtime, TensorRT.
Interview Process
2 Technical Rounds
Apply NowJob Type
Full time contract
Category
Machne learning developer
Other Jobs at UltraGenius
Fast Apply
UltraGenius
Junior AI Specialist- LLMs and RAG
We're on the hunt for a dynamic AI Specialist with specialized skills in Large Language Models (LLMs) and Retrieval-Augmented Generation (RAG).
What’s in It for you:
-
Lead the charge in AI: Work on developing state-of-the-art LLMs and RAG systems. -
Collaborate with the best: Collaborate with the industry's elite, a team of seasoned experts who are as passionate about innovation as you are. -
Tackle unprecedented challenges: Engage with complex problems that will test your skills and push the boundaries of what's possible.
We’re looking for someone who:
- Has deep, commanding expertise in LLMs and RAG.
- Boasts a solid foundation in the principles of generative AI.
- Is proficient in Python and AI development tools.
Send resumes with your best genAI projects, and apply now to become part of a fast-paced, high-delivering team where your talent and experience are highly valued and recognized.
Fast Apply
UltraGenius
AI Specialist- LLMs and RAG
If you're among the sharpest minds in AI and ready for a career that's as challenging as it is rewarding, we want to hear from you!
We're on the hunt for a dynamic AI Specialist with specialized skills in Large Language Models (LLMs) and Retrieval-Augmented Generation (RAG).
What’s in It for you:
-
Lead the charge in AI: Work on developing state-of-the-art LLMs and RAG systems. -
Collaborate with the best: Collaborate with the industry's elite, a team of seasoned experts who are as passionate about innovation as you are. -
Tackle unprecedented challenges: Engage with complex problems that will test your skills and push the boundaries of what's possible.
We’re looking for someone who:
- Has deep, commanding expertise in LLMs and RAG.
- Boasts a solid foundation in the principles of generative AI.
- Is proficient in Python and AI development tools.
Send resumes with your best genAI projects, and apply now to become part of a fast-paced, high-delivering team where your talent and experience are highly valued and recognized.
ultraGenius: a fresh start
Ultragenius has given me a new start in my life. I always wanted to start my gig and work as a freelancer for good clients, and these guys have helped me accomplish my long term goal in only 3rd year of my career. They are highly supportive and helpful. I am grateful that I got a chance to work with them.
Mahesh Inder, Reactjs DevExcellent guys
Shailesh is excellent person to work with if your startup or organisation is finding talent for your web3 product and he is also helping developers and builders in web3 to find great team to work remotely. so he is creating win-win situation for both.
Mitul Gajera, web3