Skip to content

Conversation

@parambole
Copy link
Collaborator

@parambole parambole commented Apr 25, 2025

Relaxed the condition or tensorflow dependency while still maintaining the minimum requirement

Issue:

  Attempting uninstall: jaxlib
    Found existing installation: jaxlib 0.6.1.dev20250425
    Uninstalling jaxlib-0.6.1.dev20250425:
      Successfully uninstalled jaxlib-0.6.1.dev20250425```

@parambole parambole requested a review from entrpn April 25, 2025 23:17
@parambole parambole changed the title Fix 413751357 updating orbax version which is lowering the JAX version Fix 413751357 updating version of depdendency which is lowering the JAX version Apr 25, 2025
@parambole parambole force-pushed the parambole/413751357 branch from f0aec4a to fccb25a Compare April 25, 2025 23:38
@parambole parambole merged commit e7d1f13 into main Apr 26, 2025
2 of 3 checks passed
coolkp added a commit that referenced this pull request Apr 28, 2025
* Add compilation metrics, parse step time in e2e, add name to train partial etc (#166)

* Add compilation metrics, script to assert metrics, add conditional import of hugginface_hub and update huggingface hub

Signed-off-by: Kunjan patel <kunjanp@google.com>

* Formatting

Signed-off-by: Kunjan patel <kunjanp@google.com>

* Fix pylint issues

Signed-off-by: Kunjan patel <kunjanp@google.com>

---------

Signed-off-by: Kunjan patel <kunjanp@google.com>

* Fix 413751357 updating version of depdendency which is lowering the JAX version (#167)

* Fix 413751357 updating orbax version which is lowering the JAX version

* Updating the right dependency

---------

Signed-off-by: Kunjan patel <kunjanp@google.com>
Co-authored-by: Param Bole <parambole@google.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants