5 months ago
Remote, United States +2 moreMid Level
Base Salary
$165k - $259k/yr
Responsibilities
- Profile and optimize existing PyTorch and JAX implementations without compromising model accuracy, reproducibility, or scientific robustness.
- Develop with CUDA, Triton, Warp, and related frameworks to accelerate performance-critical code.
- Collaborate with NVIDIA to represent Achira’s needs and implement NVIDIA tooling in the company’s environment.
- Work with scientists to identify high-impact optimization opportunities and collaborate with teams in San Francisco and New York.
Requirements
- At least two years of professional experience in GPU optimization.
- Deep understanding of GPU programming fundamentals.
- A strong track record of observable optimization artifacts, such as GitHub work.
- Experience collaborating on software projects across multi-person teams.
- Experience with multi-cloud distributed compute systems is preferred.
- Experience working with distributed teams across multiple sites is preferred.
- Experience with equivariant architectures operating on 3-D point clouds is preferred.
Benefits
- Preferred relocation to San Francisco or New York City, with full-time remote considered for exceptional candidates willing to travel frequently.
- Travel to conferences and corporate on-site activities is part of the role.
- Collaboration includes travel to San Francisco and New York working groups.
Tech Stack
Categories
About Achira
Achira builds atomistic foundation simulation models that simulate molecular interactions to support drug discovery and preclinical research for biotech and pharmaceutical teams. The privately held company, founded in 2024 and based in San Francisco, offers its technology as software and research collaborations. Its work combines AI with large-scale physics-based simulation to accelerate target validation and lead optimization.