Reka is an AI research and product company that develops models to benefit humanity, organizations, and enterprises.
Member of Technical Staff – GPU Performance Engineer
Location
United States
Posted
73 days ago
Salary
0
Seniority
Lead
Job Description
Member of Technical Staff – GPU Performance Engineer
Reka AI
• Design and implement improvements to training infrastructure • Directly contribute to technical decisions optimizing performance of models • Work on post-training processes, including reinforcement learning and fine-tuning • Improve the efficiency and scalability of model serving infrastructure
Job Requirements
- Strong engineering skills with fluency in Python and PyTorch (or other frameworks)
- Proven experience implementing and training large deep learning models
- Experience writing and debugging low-level GPU code (CUDA, C++)
- Experience scaling up GPU jobs using large-scale compute clusters (e.g., Slurm or Kubernetes)
- Demonstrated ability to analyze and optimize the performance of GPU-accelerated workloads, including profiling, identifying bottlenecks, and implementing performance tuning techniques
Benefits
- Five weeks of paid leave to recharge
- Comprehensive healthcare benefits (including vision and dental)
- Additional perks that support your well-being
- Visa assistance, including H1B and OPT transfers, for US employees
Related Guides
Related Job Pages
More Software Engineer Jobs
• Develop and implement new features in our systems using Java and its core technologies.
• Lead the design and development of comprehensive features for the Glider.fi flagship product, ensuring they are scalable, secure, and efficient. • Tackle unique challenges related to declarative trading and vertical solutions, bringing creative and well-architected approaches to complex problems. • Engage with every layer of the crypto/DeFi stack, from backend infrastructure to on-chain swaps and frontend Web3 UX, ensuring seamless integration and functionality.
Senior Software Developer, Web Architect
Advancis Software & ServicesA global leader in open architecture command & control system software for integrated security & building management.
• Maintenance of our existing web application “Advision” for corporate organization (PHP, JavaScript, MySQL, Elasticsearch) • Redevelopment of “Advision” for the “Advancis Open Platform (AOP)” as a modern server-/client solution for the cloud (Node.js, TypeScript, Angular) • Further development of the web client for our incident management system “WinGuard” (TypeScript, Angular) • Further development of mobile apps for iOS and Android (TypeScript, Angular, CapacitorJS) • Maintenance and extension of our JavaScript SDK (TypeScript) • Design, architecture and development of software features • Refactoring and optimization of existing code • Conducting code reviews as part of team pull requests • Participation in our architecture guild • Analysis and resolution of bugs
Software Engineer II - Infra
Bill.comBill.com offers digital bill-paying services that are used to process more than $25 billion in payments annually. Headquartered in Palo Alto, California, Bill.com has offered work-
Design and maintain internal tools to enhance the software development lifecycle, collaborate with teams to improve efficiency and code quality, and contribute to a culture of continuous learning and technical knowledge sharing.
