Add a dependence for exp04

machine-discovery / deer

Parallelizing non-linear sequential models over the sequence length

BSD 3-Clause "New" or "Revised" License

40 stars 1 forks source link

Closed zqigolden closed 11 months ago

zqigolden commented 11 months ago

I installed dear in a new conda env and tensorboardX was required for exp 04.

mfkasim1 commented 11 months ago

Could you please check if using >= in the tensorboardX dependency, would it force updating the JAX's version?

zqigolden commented 11 months ago

tensorboardX relies on the following libraries:

requirements = [
    'numpy',
    'packaging',
    'protobuf>=3.20',
]

So it should have no impact on JAX.