python311Packages.optax: fix build and bump version

[stephen-huan]

Description of changes

2c3b681 (an update from 0.1.7 to 0.2.1) breaks the build as google-deepmind/optax@f0dfdaa removed setup.py in 0.1.9. This PR switches to the pyproject format and also bumps the version. Note that this PR depends on #297146.

Diff: google-deepmind/optax@refs/tags/v0.2.1...v0.2.2

Changelog: https://github.com/deepmind/optax/releases/tag/v0.2.2

cc @mweinelt

Things done

• Built on platform(s)
  • x86_64-linux
  • aarch64-linux
  • x86_64-darwin
  • aarch64-darwin
• For non-Linux: Is sandboxing enabled in nix.conf? (See Nix manual)
  • sandbox = relaxed
  • sandbox = true
• Tested, as applicable:
  • NixOS test(s) (look inside nixos/tests)
  • and/or package tests
  • or, for functions and "core" functionality, tests in lib/tests or pkgs/test
  • made sure NixOS tests are linked to the relevant packages
• [] Tested compilation of all packages that depend on this change using nix-shell -p nixpkgs-review --run "nixpkgs-review rev HEAD". Note: all changes have to be committed, also see nixpkgs-review usage
• Tested basic functionality of all binary files (usually in ./result/bin/)
• 24.05 Release Notes (or backporting 23.05 and 23.11 Release notes)
  • (Package updates) Added a release notes entry if the change is major or breaking
  • (Module updates) Added a release notes entry if the change is significant
  • (Module addition) Added a release notes entry if adding a new NixOS module
• Fits CONTRIBUTING.md.

---

Add a 👍 reaction to pull requests you find important.

[stephen-huan]

Superseded by #297146.