SWE - Distributed
AchiraSan Francisco, California, United States • New York, New York, United States
Posted October 7, 2025
Full Time
About the Company
Achira is building best-in-class foundation models to solve the most challenging problems in simulation for drug discovery and beyond.
Job Description
Join a world-class team of scientists, ML researchers, and engineers to reshape the future of drug discovery. Work on cutting-edge ML infrastructure at frontier scale, and own impactful work end-to-end.
Requirements
- 3+ years of experience in building or working with distributed computing frameworks (e.g., Ray, Dask, Celery)
- Good grasp of parallel computing, job scheduling, and resource management
- Comfortable identifying and resolving performance issues in distributed systems (profiling, bottlenecks, network overhead)
- Experience with cloud compute platforms (AWS, GCP, Azure) and cluster orchestration (Kubernetes, Slurm)
- Familiarity with popular ML frameworks (PyTorch, TensorFlow, or JAX) and MLOps best practices such as model deployment and GPU performance monitoring
Benefits
- Competitive salary
- Benefits package (not specified)
- Opportunity to work with a world-class team