Mentions légales du service

Skip to content
Snippets Groups Projects
Verified Commit f7a13dff authored by ANDREY Paul's avatar ANDREY Paul
Browse files

Fix Jax dependency specification.

parent 9113e8ad
No related branches found
No related tags found
No related merge requests found
Pipeline #802908 passed
Stage: test
    Hide whitespace changes
    Inline Side-by-side
    Showing
    with 2 additions and 2 deletions
    pyproject.toml
    + 2
    − 2
    View file @ f7a13dff
    ......@@ -50,7 +50,7 @@ all = [ # all non-tests extra dependencies
    "dm-haiku == 0.0.9",
    "functorch",
    "grpcio >= 1.45",
    "jax[cpu] == 0.4",
    "jax[cpu] >= 0.4, < 0.5",
    "opacus ~= 1.1",
    "protobuf >= 3.19",
    "tensorflow ~= 2.5",
    ......@@ -67,7 +67,7 @@ grpc = [
    ]
    haiku = [
    "dm-haiku == 0.0.9",
    "jax[cpu] == 0.4", # NOTE: GPU support must be manually installed
    "jax[cpu] >= 0.4, < 0.5", # NOTE: GPU support must be manually installed
    ]
    tensorflow = [
    "tensorflow ~= 2.5",
    ......
    0% or .
    You are about to add 0 people to the discussion. Proceed with caution.
    Finish editing this message first!
    Please register or to comment