Differentiable astrophysics at scale: solving and differentiating ODE ensembles on the GPU
यह शोध पत्र GRADSOLVE का परिचय देता है, जो एक JAX लाइब्रेरी है जो खगोलीय मॉडलों के साधारण अवकल समीकरणों (ordinary differential equations) के समूहों को हल करने और उन्हें विभेदित (differentiate) करने के लिए GPU समानांतरता (parallelism) का लाभ उठाती है, जिससे सीरियल CPU कार्यान्वयन की तुलना में 1,500 गुना तक की गति वृद्धि प्राप्त होती है और लाखों प्रक्षेप पथों (trajectories) के नियमित ग्रेडिएंट-आधारित विश्लेषण को सक्षम करने के लिए मौजूदा JAX ODE सॉल्वर से काफी बेहतर प्रदर्शन होता है।