Search for supersymmetry in events with $b$-tagged jets and missing transverse momentum in $pp$ collisions at $\sqrt{s}=13$ TeV with the ATLAS detector