[Remote] Sr. Machine Learning Researcher

Note: The job is a remote job and is open to candidates in USA. AKASA is a leading provider of generative AI solutions for the healthcare revenue cycle, aiming to transform healthcare operations. As a Senior Machine Learning Researcher, you will develop advanced AI models to enhance efficiency and precision in healthcare workflows, collaborating with a team of experts to drive impactful research and product innovation. Responsibilities • Drive Applied Research: Lead the design, training, and evaluation of large language models to solve healthcare-specific challenges, advancing the state of the art in clinical Natural Language Understanding • Leverage Human-in-the-Loop Feedback: Work closely with cross-functional teams to integrate Human-in-the-Loop data, using it to guide model improvements and explore new methods for optimizing performance • Collaborate Across Teams: Partner with healthcare experts and other stakeholders to integrate qualitative insights, ensuring models align with real-world needs and deliver meaningful results • Stay on the Cutting Edge: Regularly evaluate advancements in ML to determine their relevance to our work, maintaining AKASA’s leading edge in responsible, high-impact healthcare AI • Contribute to Broader Impact: Publish and share research findings in the broader AI community, helping to advance healthcare applications of AI through peer-reviewed publications Skills • A passion for tackling meaningful healthcare challenges and leveraging AI to make a positive impact on the healthcare system • Proven experience training and fine-tuning large language models, with expertise in model architecture, optimization techniques, and performance evaluation • Experience with modeling in the healthcare domain, clinical natural language understanding, and healthcare data and data standards (e.g., EDI, FHIR) is strongly preferred • Ph.D. or equivalent industry experience in fields related to machine learning, natural language processing, computer vision, computer science, electrical engineering, statistics, mathematics, optimization, or data science, plus 2-3 years of experience • Peer-reviewed publications in top AI conferences (e.g., NeurIPS, ICML, ICLR, CVPR, ICCV, ECCV, ACL, NAACL, and EMNLP) strongly preferred • Strong programming skills in Python, with proficiency in deep learning frameworks and tools such as PyTorch, TensorFlow, PyTorch Lightning, Hugging Face Transformers, and Kubernetes/Kubeflow • Experience with cloud platforms (AWS, GCP) and multi-GPU environments • A track record of translating research into real-world applications, with experience in rapid prototyping, experimentation, and writing production-ready code • Ability to work effectively in a cross-functional, fast-paced environment, with strong written and verbal communication skills Benefits • Unlimited paid time off (PTO) • Expansive coverage for health, dental, and vision • Employer contribution to Health Savings Accounts (HSA) • Generous parental leave policy • Full employee coverage for life insurance • Company-paid holidays • 401(K) plan Company Overview • Healthcare revenue cycle teams are strained. Volumes are rising, margins are constricting, and hiring and retaining enough staff is impossible. It was founded in 2019, and is headquartered in South San Francisco, CA, US, with a workforce of 51-200 employees. Its website is Company H1B Sponsorship • AKASA has a track record of offering H1B sponsorships, with 5 in 2024, 1 in 2023, 8 in 2022, 2 in 2021. Please note that this does not guarantee sponsorship for this specific role. Apply tot his job