Code

Upgrading Jax version#11

Code

Open

kew96 wants to merge 5 commits intotoshikwa:mastertoshikwa/rljax:masterfrom kew96:masterkew96/rljax:masterCopy head branch name to clipboard

+30-28

Commits

Commits on Sep 29, 2021

• Switch from jax.experiment.optix to optax

  kew96

  committed
  a008823

  Copy full SHA for a008823

• Fix version of optax

  kew96

  committed
  70d5035

  Copy full SHA for 70d5035

• Bump Haiku version to support latest jax

  kew96

  committed
  57b010f

  Copy full SHA for 57b010f

• Change imports to optax

  kew96

  committed
  dacbe1f

  Copy full SHA for dacbe1f

• Added jax and jaxlib to requirements.txt

  kew96

  committed
  f80998b

  Copy full SHA for f80998b