Because there are languages and libraries that are a better fit for some use cases like for example PyTorch or JAX. The latter has been used in for example the research below.<p>A Differentiable Neural-Network Force Field for Ionic Liquids. (J. Chem. Inf. Model. 2022)<p>H. Montes-Campos, J. Carrete, S. Bichelmaier, L. M. Varela, and G. K. H. MadsenCorrelation Tracking: Using simulations to interpolate highly correlated particle tracks. (Phys. Rev. E. 2022)<p>E. M. King, Z. Wang, D. A. Weitz, F. Spaepen, and M. P. BrennerOptimal Control of Nonequilibrium Systems Through Automatic Differentiation.
M. C. Engel, J. A. Smith, and M. P. BrennerGraph Neural Networks Accelerated Molecular Dynamics. (J. Chem. Phys. 2022)
Z. Li, K. Meidani, P. Yadav, and A. B. FarimaniGradients are Not All You Need.<p>L. Metz, C. D. Freeman, S. S. Schoenholz, and T. KachmanLagrangian Neural Network with Differential Symmetries and Relational Inductive Bias.<p>R. Bhattoo, S. Ranu, and N. M. A. KrishnanEfficient and Modular Implicit Differentiation.<p>M. Blondel, Q. Berthet, M. Cuturi, R. Frostig, S. Hoyer, F. Llinares-López, F. Pedregosa, and J.-P. VertLearning neural network potentials from experimental data via Differentiable Trajectory Reweighting. (Nature Communications 2021)<p>S. Thaler and J. ZavadlavLearn2Hop: Learned Optimization on Rough Landscapes. (ICML 2021)<p>A. Merchant, L. Metz, S. S. Schoenholz, and E. D. CubukDesigning self-assembling kinetics with differentiable statistical physics models. (PNAS 2021)<p>C. P. Goodrich, E. M. King, S. S. Schoenholz, E. D. Cubuk, and M. P. Brenner<p>Some threads discussing the merits of JAX, PyTorch and Julia:<p><a href="https://news.ycombinator.com/item?id=30349687" rel="nofollow">https://news.ycombinator.com/item?id=30349687</a><p><a href="https://news.ycombinator.com/item?id=29682507" rel="nofollow">https://news.ycombinator.com/item?id=29682507</a>
<a href="https://www.reddit.com/r/MachineLearning/comments/shsfkm/d_current_state_of_jax_vs_pytorch/" rel="nofollow">https://www.reddit.com/r/MachineLearning/comments/shsfkm/d_c...</a>
<a href="https://www.reddit.com/r/MachineLearning/comments/st8b11/d_should_we_be_using_jax_in_2022/" rel="nofollow">https://www.reddit.com/r/MachineLearning/comments/st8b11/d_s...</a><p><a href="https://www.assemblyai.com/blog/why-you-should-or-shouldnt-be-using-jax-in-2022/" rel="nofollow">https://www.assemblyai.com/blog/why-you-should-or-shouldnt-b...</a><p><a href="https://archive.ph/FL6hs" rel="nofollow">https://archive.ph/FL6hs</a>