{"id":747328,"url":"https://alion.io/job/achira-machine-learning-research-engineer-mlre-gpus","title":"Machine Learning Research Engineer (MLRE) - GPUs","company":{"id":669603,"name":"Achira","domain":"achira.ai","url":"https://alion.io/company/achira","size_band":"201-500","is_staffing_agency":false,"employer_type":"direct","is_intermediary":false,"listed_via":null,"ats_vendor":"Ashby","truth_index":{"grade":"B","score":75,"open_postings":6,"ghost_share":0,"stale_share":1,"repost_share":0,"time_to_fill_p50_days":null,"computed_at":"2026-10-02T05:45:00Z"}},"role":"AI/ML","role_family":"AI/ML","seniority":"junior","employment_type":"full_time","work_mode":"hybrid","remote_scope":null,"remote_scope_basis":null,"remote_working_hours":null,"hiring_geo_confidence":"structured","locations":["San Francisco, United States","New York, United States"],"countries":["US"],"hiring_countries":[],"hiring_countries_total":0,"salary":{"min":164638,"max":259000,"currency":"USD","period":"year","gross":null,"usd_annual":259000},"salary_estimate":null,"experience_years_min":2,"visa_sponsorship":false,"relocation_package":false,"has_equity":false,"technologies":[{"name":"CUDA","optional":false},{"name":"CUDA Toolkit","optional":false},{"name":"GitHub","optional":false},{"name":"JAX","optional":false},{"name":"Machine Learning","optional":false},{"name":"PyTorch","optional":false},{"name":"Triton","optional":false},{"name":"World Models","optional":false}],"status":"live","first_seen_at":"2026-04-29T14:28:53Z","employer_posted_date":"2026-04-29","last_verified_at":"2026-10-02T04:55:58Z","board_verified":true,"closed_at":null,"days_open":156,"trust":{"level":"stale","repost_count":0,"flags":["stale"],"days_open":155},"description":"Why Achira\nAt Achira, we are building a team of world-class scientists, ML researchers, and engineers to work together to move beyond the beaten path in drug discovery. We are actively exploring the next frontier of model architectures for AI x Chemistry: developing world models for the physical microcosm. Our goal is to make biology at the molecular level something that can be learned, predicted, and designed.\nAt Achira, you’ll operate at the frontier scale of massive compute, massive data, and massive ambition. You’ll own impactful work end-to-end, from ideation to architecture to deployment on distributed infrastructure. We are a well-funded, talent-dense organization that values rigor, speed, execution, and an ownership mindset. We’re looking for new members who share our sense of relentless urgency and are natural collaborators who value team success.\nAbout the Role\nWe're looking for a rare individual who thrives at the intersection of cutting-edge deep learning architectures and high-performance computing. You will help shape the future of molecular machine learning by engineering high-efficiency implementations of advanced architectures for foundation simulation models, accelerating simulations to the limits of the hardware while maintaining fidelity to the underlying physics.\nWhile we prefer candidates willing to relocate to San Francisco or New York City, we are willing to consider full-time remote candidates of exceptional talent who are willing to travel frequently to our two office sites. Travel is part of all roles at Achira, both to conferences and corporate on-site activities.\nWhat You’ll Do\nTake existing PyTorch and JAX, profile and optimize it without compromising model accuracy, reproducibility, and robustness with respect to scientific objectives.\n\nDevelop with frameworks like CUDA, Triton, Warp, etc. to accelerate performance critical code sections.\n\nLiaise with NVIDIA to represent our needs and implement their tooling in our environment.\n\nWork day-to-day with scientists to identify areas of greatest impact, including travel to our SF and NY working groups to collaborate.\n\nAbout You\nEngineer with at least two years professional experience in GPU optimization.\n\nDeep understanding of GPU programming fundamentals.\n\nSolid track record of observable artifacts (e.g., GitHub) showing optimization work.\n\nExperience collaborating on software projects across multi-person teams.\n\nNice to Have\nEven if you hit none of these bonus features, we encourage you to apply!\nExperience working with multi-cloud distributed compute systems.\n\nExperience working with multi-site distributed company team.\n\nExperience working with equivariant architectures that operate on 3-D point clouds.","description_format":"text","description_chars":2729,"description_truncated":false,"requirements":{"experience_years_min":2,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":null,"security_clearance":false,"languages":[]},"benefits":[],"hiring_locations":[{"name":"United States","iso":"US","kind":"country"}],"hiring_excludes":[],"relocation_offered":true,"industries":["Drug Discovery","Drug Discovery AI"],"lifecycle":[{"event":"open","at":"2026-09-11T14:36:41Z"}],"liveness":{"score":5,"band":"cold","label":"Long shot","p_open":1,"p_active":0.18,"p_room":0.28,"age_days":155,"expected_fill_days":30,"reasons":["conf:0","win:tail","crowd:junior"],"computed_at":"2026-10-02T05:45:00Z"},"pay":{"stated_usd_annual":259000,"is_top_pay":true},"html_url":"https://alion.io/job/achira-machine-learning-research-engineer-mlre-gpus","json_url":"https://alion.io/job/achira-machine-learning-research-engineer-mlre-gpus.json","meta":{"generated_at":"2026-10-03T04:09:55Z","cache_seconds":300,"methodology":"https://alion.io/methodology","terms":"https://alion.io/terms","contact":"https://alion.io/contact","api":"https://alion.io/developers","usage":{"tier":"crawler","counted_by":"address","units_charged":1,"used_today":4268,"day_limit":5000,"remaining_today":732,"minute_limit":60,"resets_at":"2026-10-04T00:00:00Z"}}}