Context. Turbulent transport of chemical elements in radiative zones of stars is considered in current stellar evolution codes thanks to phenomenologically derived diffusion coefficients. Recent local numerical simulations suggest that the coefficient for radial turbulent diffusion due to radial differential rotation satisfies D t 0.058κ/Ri, in qualitative agreement with the model of Zahn (1992, A&A, 265, 115). However, this model does not apply (i) when differential rotation is strong with respect to stable thermal stratification or (ii) when chemical stratification has a significant dynamical effect, a situation encountered at the outer boundary of nuclear-burning convective cores. Aims. We extend our numerical study to consider the effects of chemical stratification and of strong shear, and compare the results with prescriptions used in stellar evolution codes. Methods. We performed local, direct numerical simulations of stably stratified, homogeneous, sheared turbulence in the Boussinesq approximation. The regime of high thermal diffusivities, typical of stellar radiative zones, is reached thanks to the so-called smallPéclet-number approximation, which is an asymptotic development of the Boussinesq equations in this regime. The dependence of the diffusion coefficient on chemical stratification was explored in this approximation. Results. Maeder's extension of Zahn's model in the strong-shear regime (Maeder 1995, A&A, 299, 84) is not supported by our results, which are better described by a model found in the geophysical literature. As regards the effect of chemical stratification, our quantitative estimate of the diffusion coefficient as a function of the mean gradient of mean molecular weight leads to the formula D t 0.45κ(0.12 − Ri µ )/Ri, which is compatible in the weak-shear regime with the model of Maeder & Meynet (1996, A&A, 313, 140) but not with Maeder's (1997, A&A, 321, 134).