ai-helpers-jax-utils

Dependencies (6)

• jax <0.5.0,>=0.4.28
• jax-dataclasses <2.0.0,>=1.6.0
• matplotlib <4.0.0,>=3.9.0
• optax <0.3.0,>=0.2.2
• ordered-set <5.0.0,>=4.1.0
• tqdm <5.0.0,>=4.66.4

Dependencies (6)

• jax <0.5.0,>=0.4.28
• jax-dataclasses <2.0.0,>=1.6.0
• matplotlib <4.0.0,>=3.9.0
• optax <0.3.0,>=0.2.2
• ordered-set <5.0.0,>=4.1.0
• tqdm <5.0.0,>=4.66.4

Dependencies (6)

• jax <0.5.0,>=0.4.28
• jax-dataclasses <2.0.0,>=1.6.0
• matplotlib <4.0.0,>=3.9.0
• optax <0.3.0,>=0.2.2
• ordered-set <5.0.0,>=4.1.0
• tqdm <5.0.0,>=4.66.4