JSFiddle - React, Tailwind, and code Playground

HTML

<table tabindex=0>
  <tr><td>R0C0
      <td>R0C1 <input>
      <td>R0C2
  <tr><td>R1C0 <input>
      <td>R1C1 
      <td>R1C2
</table>
<input>

CSS

td {
  border: 1px solid #ddd;
  padding: 1em;
}

input:focus {
  background:yellow
}

table:focus {
  background:yellow
}

JavaScript

$('td,th').on('focus',
  function() {
    $(this).closest('table').focus()
  }
)