8000 Commits · google/rax · GitHub

More Web Proxy on the site http://driver.im/

Commits

Commits on May 8, 2025

[rax] avoid use of deprecated `jax.util` APIs

Jake VanderPlas
authored and
Rax Developers
committed

Commits on Mar 5, 2025

Update references to JAX's GitHub repo

jakeharmon8
authored and
Rax Developers
committed

Commits on Feb 5, 2025

Relax numerical precision tolerance in unit tests and doctests.

rjagerman
authored and
Rax Developers
committed

Commits on Jan 3, 2025

Bump version to v0.4.0

Nikitas Rontsis
authored and
Rax Developers
committed

Commits on Oct 28, 2024

Remove use of `initial` argument to `jax.nn.softmax` and `jax.nn.log_softmax`

Jake VanderPlas
authored and
Rax Developers
committed

Commits on Oct 16, 2024

Expose `normalize_probabilities` as a good normalization for `SoftmaxLoss`
Rax Developers
committed

Commits on Jul 6, 2024

Align documentation with up-to-date code.

rjagerman
authored and
Rax Developers
committed

Commits on Jun 26, 2024

Add safe_reduce to rax.utils for loss/metric implementations outside of rax.
Rax Developers
committed

Commits on Feb 29, 2024

Fix Array import in segment_utils

rjagerman
authored and
Rax Developers
committed

Commits on Feb 16, 2024

Fix pytype and clean up types across codebase.

rjagerman
authored and
Rax Developers
committed

Commits on Feb 7, 2024

[rax] add a light discount option for dcg2_lambdaweight

xuanhuiwang
authored and
Rax Developers
committed
[rax] refactor the lambda weight to reduce the boilerplate.

xuanhuiwang
authored and
Rax Developers
committed

Commits on Jan 22, 2024

internal_change

TerenceLiu4444
authored and
Rax Developers
committed
Drop python 3.8 checks and add python 3.11 checks.

rjagerman
authored and
Rax Developers
committed
Internal change.

rjagerman
authored and
Rax Developers
committed

Commits on Jan 12, 2024

Internal change.

TerenceLiu4444
authored and
Rax Developers
committed

Commits on Oct 18, 2023

[LSC] change uses of jax.random.KeyArray and jax.random.PRNGKeyArray to jax.Array

Jake VanderPlas
authored and
Rax Developers
committed

Commits on Oct 9, 2023

Ignore incorrect type annotations related to jax dtypes

Jake VanderPlas
authored and
Rax Developers
committed

Commits on Oct 6, 2023

Ignore incorrect type annotations related to jax dtypes

Jake VanderPlas
authored and
Rax Developers
committed

Commits on Sep 25, 2023

[LSC] Ignore incorrect type annotations related to jax.numpy APIs

Jake VanderPlas
authored and
Rax Developers
committed

Commits on Jun 30, 2023

Fix typo in `opa_metric` docstring.

rjagerman
authored and
Rax Developers
committed
Add Ordered Pair Accuracy (OPA) metric.
Rax Developers
committed

Commits on Jun 29, 2023

Making tests accommodate both FrozenDict and regular dict variables, as Flax will default to returning regular dicts in the future.

chiamp
authored and
Rax Developers
committed

Commits on Jun 7, 2023

Merge pull request #105 from thomas-a-neil:main
Rax Developers
committed

Commits on Jun 2, 2023

fix typo in loss function docstring
thomas-a-neil
committed
81bc1bd

Commits on May 17, 2023

Making tests accommodate both FrozenDict and regular dict variables, as Flax will default to returning regular dicts in the future.

chiamp
authored and
Rax Developers
committed
Fix PyPI publish pipeline.

rjagerman
authored and
Rax Developers
committed
Bump version to v0.3.0

rjagerman
authored and
Rax Developers
committed
Add example using segmented loss and metrics.

rjagerman
authored and
Rax Developers
committed
Fix `segment_{sum/max}` to appropriately sanitize NaN values.

rjagerman
authored and
Rax Developers
committed

Commits on May 5, 2023

Update general Rax info.

rjagerman
authored and
Rax Developers
committed
Fix readthedocs configuration

rjagerman
authored and
Rax Developers
committed

Commits on May 4, 2023

Format example code.

rjagerman
authored and
Rax Developers
committed
Format code.

rjagerman
authored and
Rax Developers
committed
Clean up docstrings.

rjagerman
authored and
Rax Developers
committed

0