SWE - Distributed
Achira· Software Engineering
Hybrid (San Francisco Office), Hybrid (New York Office)$164.6k - $259k + equity + bonusFull-timePosted 11mo ago
About the role
Achira is building foundation models for simulation in drug discovery, focusing on distributed ML data generation, training, and fine-tuning pipelines across large-scale distributed systems. The role involves architecting and building infrastructure, optimizing performance and cost, and advancing ML workflows across multiple cloud and cluster environments.
Responsibilities
- Architect & Build: Design, implement, and optimize distributed compute infrastructure for ML data processing, training, and fine-tuning.
- Optimize & Monitor: Improve cluster observability, scheduling, and resource utilization (CPU/GPU/TPU).
- Compute Efficiency: Research and implement cost-efficient compute solutions (spot instances, auto-scaling, multi-cloud strategies).
- Tooling: Develop tools for monitoring, debugging, and performance tuning of large-scale ML workloads.
- Collaboration: Collaborate with ML engineers to accelerate training pipelines and reduce bottlenecks.
- Innovation: Stay current with emerging technologies in distributed computing (e.g., Ray, Kubernetes, Spark, Slurm) and apply them strategically.
Qualifications
- Experience in building or working with distributed computing frameworks (e.g., Ray, Dask, Celery)
- Strong grasp of parallel computing, job scheduling, and resource management
- Experience 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 ML frameworks (PyTorch, TensorFlow, or JAX) and MLOps best practices such as model deployment and GPU performance monitoring
Skills
Required:Distributed computing frameworksParallel computing & schedulingCloud platformsKubernetes / SlurmML frameworks
Benefits
- Independent Contributor
- Equity offered
- Bonus offered
Last updated Jul 15, 2026. Interested? Apply on the company site.
About Achira
Building foundation simulation models for drug discovery
Achira is building atomistic foundation simulation models to power the future of drug discovery.
drug discoverysimulation modelsmachine learningquantum chemistrybiomolecular applications
More open roles at Achira
- People Operations Manager1mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Operational
- ML Research Scientist (MLRS) - Representation Learning for Molecular AI3mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Machine Learning
- ML Research Scientist (MLRS) - Generative AI3mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Machine Learning
- Machine Learning Research Engineer (MLRE) - Research5mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Machine Learning
- Machine Learning Research Engineer (MLRE) - Workflows/Systems5mo agoHybrid (San Francisco Office), Hybrid (New York Office) · Machine Learning