Temp testing automation, please ignore#145

Open

RerRayne wants to merge 57 commits intotest_638428288from

Commits on May 30, 2024

PUBLIC: Add `predecessor_pointers_to_permutation_matrix` and `permutation_matrix_to_predecessor_pointers` methods to probing.

CLRSDev
authored and
copybara-github
committed
PUBLIC: Add an additional field to the `quicksort` probing, update its specifications, and add a `track_max_steps` flag to the sampler. This flag will enable or disable length tracking for padding.

CLRSDev
authored and
copybara-github
committed

Commits on May 31, 2024

Commits on Jun 6, 2024

Commits on Jun 9, 2024

[JAX] Update users of jax.tree.map() to be more careful about how they handle Nones.

hawkinsp
authored and
copybara-github
committed

Commits on Jun 14, 2024

Internal

RerRayne
authored and
copybara-github
committed

Commits on Jun 21, 2024

Improving quality of huggingface generators
mcleish7
committed

Commits on Jun 26, 2024

Apply suggestions from code review

mcleish7
and
grenlayk
authored

Commits on Jun 27, 2024

Add `ml_collections` to `requirements.txt`.

RerRayne
authored and
copybara-github
committed

Commits on Jun 28, 2024

Made 'text' a permanent field and switched to uniform random sampling instead of exactly equal sampling.
mcleish7
committed

Commits on Jun 30, 2024

Combining the infinite and finite generators and general comment formatting fixes
mcleish7
committed

Commits on Jul 1, 2024

Fix minor linter and codestyle issues
grenlayk
committed

Commits on Jul 4, 2024

Modify tests and add copyright
grenlayk
committed
Fix typo in tests params
grenlayk
committed
Added script to generate json for all algorithms.

RerRayne
authored and
copybara-github
committed

Commits on Jul 5, 2024

Merge pull request #135 from mcleish7/clrs-text-hf
RerRayne
authored
Add support for generating train and val datasets using this script.

RerRayne
authored and
copybara-github
committed
Add clrs_text to __init__.py

RerRayne
authored and
copybara-github
committed

Commits on Jul 8, 2024

Merge pull request #135 from mcleish7:clrs-text-hf
copybara-github
committed

Commits on Jul 10, 2024

Commits on Jul 11, 2024

Add CLRS-Text details in the README files.

PetarV-
authored and
copybara-github
committed
Use pythonic swap for variables in sorting.py

gurux13
authored and
RerRayne
committed
Merge pull request #147 from gurux13:master
copybara-github
committed

Commits on Jul 15, 2024

Bump CLRS version to 2.0.0.

RerRayne
authored and
copybara-github
committed

Commits on Jul 18, 2024

Commits on Sep 6, 2024

Update version numbers from requirements.txt

RerRayne
authored and
copybara-github
committed

Commits on Sep 22, 2024

Log the warning only for the first on-the-fly sampler.

sinopalnikov
authored and
copybara-github
committed

Commits on Nov 13, 2024

Update run.py.

grenlayk
authored and
copybara-github
committed

Commits on Nov 27, 2024

Adds a "debug" mode to Baselines, in order to support decoding state.

PetarV-
authored and
copybara-github
committed

Commits on Dec 19, 2024

Resolve unsoundness caught by pytype --strict-none-binding.

CLRSDev
authored and
copybara-github
committed

Commits on Jan 20, 2025

[CLRS]

RerRayne
authored and
copybara-github
committed
[CLRS] Add num_decimals_in_float parameter to the data generators.

RerRayne
authored and
copybara-github
committed
[CLRS] Incorporate a collinearity check into the ConvexHull sampler to prevent generating samples that admit multiple valid paths.

RerRayne
authored and
copybara-github
committed

Commits on Jan 25, 2025

Increment minor version number for 2.0.3 release.

RerRayne
authored and
copybara-github
committed

Commits on Jan 28, 2025

Internal.

RerRayne
authored and
copybara-github
committed

Commits on Mar 20, 2025

Replace numpy cross product with hand-written version

RerRayne
authored and
copybara-github
committed

Commits on Apr 30, 2025

Internal change

h-joo
authored and
copybara-github
committed

Commits on Oct 15, 2025

Fix bug where train lengths were modified across algorithms.

RerRayne
authored and
copybara-github
committed

Commits on Dec 29, 2025

[pmap] Avoid degraded performance under the new `jax.pmap`.

danielsuo
authored and
copybara-github
committed

Commits on Jan 9, 2026

Fix JAX tree mismatch error in filter_null_grads

RerRayne
authored and
copybara-github
committed
Replace unicode escaped characters in ipynb files

AleksMat
authored and
copybara-github
committed

Commits on Feb 2, 2026

Fix flaky `losses_test`

RerRayne
authored and
copybara-github
committed

Commits on Feb 3, 2026

[pmap] Make clrs nets.py _get_first more robust under jax_pmap_shmap_merge

danielsuo
authored and
copybara-github
committed

Commits on Mar 2, 2026

[pmap] Remove `jax.config.pmap_shmap_merge`.

danielsuo
authored and
copybara-github
committed

Commits on Mar 26, 2026

[pmap] In-line definitions of `jax.device_put_sharded` and `jax.device_put_replicated`.

danielsuo
authored and
copybara-github
committed