jax

5ebbdf2c - [RFC] Enabled unused-variable in Ruff's config

[RFC] Enabled unused-variable in Ruff's config I inspected all existing instances of unused variables and either removed them or prefixed their name with with an `_`. See https://docs.astral.sh/ruff/rules/unused-variable.