Anlatan
Responsibilities
- Working with our distributed training stack to start training runs and tracking them closely.
- Contributing to our ML codebases and adding new models, algorithms and features.
- Experimenting with new model architectures and designs, and getting them production ready.
- Training and finetuning models, and adapting them to our products by taking tester feedback and iterating over the trained models.
Requirements
- Experience with pytorch, python
- Past experience with training NLP or Image models.
- Able to read cutting edge ML papers, implement them in pytorch and accurately reproduce.
- Strong knowledge of computer science fundamentals, including algorithms and data structures.
- Machine learning fundamentals
- Good problem-solving and troubleshooting skills.
Bonus Points
- Knowledge in Kubernetes and SLURM
- Distributed computing knowledge to contribute to our distributed ML training stack.
- Experience with JAX
- Experience with Triton
Originally posted on Himalayas
To apply for this job please visit himalayas.app.
About this role & career path
Working in United States
The United States of America (USA), also known as the United States (U.S.) or America, is a country primarily located in North America. It is a federal republic consisting of 50 states and a federal capital district, Washington, D.C. The 48 contiguous states border Canada to the north and Mexico to the south, with the semi-exclave of Alaska in the northwest and the archipelago of Hawaii in the Pacific Ocean. The United States also asserts sovereignty over five major island territories and various uninhabited islands in Oceania and the Caribbean. It is a megadiverse country, with the world's th
More jobs at Anlatan
Keep exploring on Get A Job.ai
Not quite the right fit? Your next opportunity is a click away.
- Browse all jobs
- More jobs by category
- Remote jobs you can do from anywhere
- Research typical pay for this role
- Set a job alert so new matches reach you first
- Upload your resume to apply faster
Hiring instead? Post a job and reach candidates searching right now.