enable grayskull update

### Solution to issue cannot be found in the documentation.

- [X] I checked the documentation.

### Issue

Here and in brax.

### Installed packages

```shell
.
```

### Environment info

```shell
.
```