About the Role
We are seeking experienced Python developers to lead the development and optimization of AI training infrastructure. The ideal candidate will have a strong background in software engineering, machine learning systems, and scalable data processing.
Responsibilities
- Design and implement core components of AI training platforms
- Optimize code for performance and scalability in distributed environments
- Collaborate with data scientists to integrate training workflows
- Develop automated testing frameworks for model training pipelines
- Troubleshoot and resolve issues in production-level AI systems
- Contribute to architectural decisions for machine learning infrastructure
- Ensure code quality through peer reviews and best practices
- Maintain documentation for systems and development processes
- Support deployment of models into production environments
- Work with large-scale datasets for training accuracy and efficiency
- Integrate third-party libraries and tools into existing pipelines
- Monitor system performance and implement improvements
- Participate in agile planning and sprint execution
- Mentor junior developers on Python and AI-related tasks
- Stay current with advancements in AI and software engineering
- Implement security measures in data handling and model training
- Ensure compliance with data privacy standards
- Refactor legacy systems for improved maintainability
- Build APIs for internal and external model access
- Coordinate with DevOps for CI/CD integration
- Evaluate new technologies for potential adoption
- Support reproducibility in training experiments
- Improve fault tolerance in distributed training jobs
- Assist in defining project scope and timelines
- Contribute to technical design documentation
Nice to Have
- Master’s degree in computer science or related field
- Experience with large language models or NLP systems
- Contributions to open-source machine learning projects
- Background in high-performance computing
- Experience with workflow management tools like Airflow
- Knowledge of model versioning systems
- Familiarity with data annotation pipelines
- Experience in regulated environments requiring audit trails
- Understanding of ethical AI principles
- Prior work in agile or lean development environments
Compensation
Competitive salary and benefits package
Work Arrangement
Hybrid work model with partial remote flexibility
Team
Collaborative engineering team focused on artificial intelligence systems
About the Role
This position plays a key role in advancing AI training capabilities through robust software engineering. Developers will work directly on systems that train and refine machine learning models, ensuring efficiency, accuracy, and scalability. The role blends deep technical work with collaboration across data science and infrastructure teams.
What We’re Looking For
We seek engineers who are passionate about building reliable, high-performance systems for AI. Candidates should have a track record of delivering production-grade software and a strong interest in the technical challenges of training artificial intelligence models at scale.
Technology Stack
Python, TensorFlow, PyTorch, Docker, Kubernetes, Git, AWS/GCP, Airflow, PostgreSQL, Redis, FastAPI, Prometheus, Grafana
Growth and Development
Engineers are encouraged to lead initiatives, propose technical improvements, and grow into architecture and mentorship roles. Regular knowledge-sharing sessions and conference participation are supported.
Work Environment
The team operates in a hybrid setup with office space available in Washington DC. Remote collaboration is facilitated through modern tools and asynchronous communication practices.
Application Process
Candidates will undergo a technical screening, followed by a coding assessment and multiple interview rounds with engineering and product team members.
Available for qualified candidates
.png?1769074621){kind=icon}