Back to all positions
🎓
GPU Developer for Probabilistic Programming
🏢HybridInternship
◆ About the role
We are seeking a talented GPU Developer Intern to help accelerate our probabilistic programming framework with GPU support. The ideal candidate has experience with CUDA, GPU programming, and parallel computing.
◆ What you'll do
- •Develop GPU acceleration for probabilistic models
- •Optimize performance for large-scale inference
- •Work at the intersection of HPC and probabilistic programming
◆ Responsibilities
- •Implement GPU kernels for common probabilistic programming operations
- •Profile and optimize existing code for GPU execution
- •Collaborate with model developers to ensure compatibility
- •Create benchmarks to measure performance improvements
- •Document GPU implementation details for future reference
◆ Requirements
- •Currently pursuing a degree in Computer Science, Engineering, or a related field
- •Experience with GPU programming (CUDA or equivalent)
- •Knowledge of parallel computing principles
- •Familiarity with Julia or Python
- •Understanding of probabilistic modeling concepts
- •Strong mathematical foundation, particularly in linear algebra
◆ Benefits
✓Competitive internship stipend
✓Flexible remote work arrangement
✓Hands-on experience with cutting-edge technology
✓Mentorship from experts in AI
✓Opportunity to contribute to open-source projects
✓Potential for full-time employment after internship
Ready to apply?
Join our team and help us build the future. We look forward to reviewing your application!
Interested in joining our team?
Explore other open positions or check back later for new opportunities.
View All Positions