PyTorch Model Optimization for Physics Simulations
Budget: $250 – $750 USD
I'm seeking an expert in PyTorch and GPU programming to optimize a complicated PyTorch simulator written in Python. This simulator is primarily used for physics-based simulations. The goal is to speed up the model computation - specifically the forward pass - by at least 25-100x.
Key Requirements:
- Strong expertise in Python and PyTorch
- Extensive experience with GPU programming
- In-depth knowledge and experience in numerical simulations and iterative solvers
- Ability to identify and resolve performance bottlenecks
- Familiarity with physics-based simulations
The optimization process will focus on the model computation phase, particularly the forward pass which is currently the most time-consuming aspect. The ideal candidate will have a proven track record of significantly improving the performance of similar projects.
Key Requirements:
- Strong expertise in Python and PyTorch
- Extensive experience with GPU programming
- In-depth knowledge and experience in numerical simulations and iterative solvers
- Ability to identify and resolve performance bottlenecks
- Familiarity with physics-based simulations
The optimization process will focus on the model computation phase, particularly the forward pass which is currently the most time-consuming aspect. The ideal candidate will have a proven track record of significantly improving the performance of similar projects.