Tests

Workflow runs for Tests

• Update documentation links and remove old requirements.

  cancelled:
  Tests#12618:Pull request#2805synchronizebycopybara-service[bot]Bot
  test_842542326

  Dec 13, 202511m 4s
  More actions for workflow run 12618

• This change introduces a pure JAX implementation of flash attention to Maxtext, designed as a drop-in replacement for the existing Pallas kernel. In this cl we set up the stage by integrating it with maxtext in fsdp mode. We have plans for further optimizations to close the gap with pallas using different techniques such as:

  cancelled:
  Tests#12617:Pull request#2793synchronizebycopybara-service[bot]Bot
  test_834764107

  Dec 13, 202513m 27s
  More actions for workflow run 12617

• Update configuration doc

  failed:
  Tests#12616:Pull request#2827synchronizebyRissyRan
  fix_doc

  Dec 13, 202555m 22s
  More actions for workflow run 12616

• Update configuration doc

  cancelled:
  Tests#12615:Pull request#2827openedbyRissyRan
  fix_doc

  Dec 13, 202510m 44s
  More actions for workflow run 12615

• Update documentation links and remove old requirements.

  failed:
  Tests#12614:Pull request#2805synchronizebycopybara-service[bot]Bot
  test_842542326

  Dec 13, 202545m 24s
  More actions for workflow run 12614

• Update documentation links and remove old requirements.

  cancelled:
  Tests#12613:Pull request#2805synchronizebycopybara-service[bot]Bot
  test_842542326

  Dec 13, 202515m 31s
  More actions for workflow run 12613

• Add citation entry for repository

  failed:
  Tests#12612:Pull request#2825synchronizebyshuningjin
  cite-repo

  Dec 13, 202557m 36s
  More actions for workflow run 12612

• Add citation entry for repository

  cancelled:
  Tests#12611:Pull request#2825synchronizebyshuningjin
  cite-repo

  Dec 13, 20251m 0s
  More actions for workflow run 12611

• Support Custom MaxText model (with vLLM engine) in RL rollouts.

  failed:
  Tests#12610:Pull request#2778synchronizebyNicoGrande
  nicogrande/maxtext-vllm-rl-integration

  Dec 13, 20251h 32m 55s
  More actions for workflow run 12610

• Update documentation links and remove old requirements.

  cancelled:
  Tests#12609:Pull request#2805synchronizebycopybara-service[bot]Bot
  test_842542326

  Dec 13, 202558m 2s
  More actions for workflow run 12609

• Refactor dockerfiles and deprecate stable_stack and jax_ai_image mode

  failed:
  Tests#12608:Pull request#2822synchronizebySurbhiJainUSC
  refactor_docker

  Dec 13, 20251h 14m 56s
  More actions for workflow run 12608

• This change introduces a pure JAX implementation of flash attention to Maxtext, designed as a drop-in replacement for the existing Pallas kernel. In this cl we set up the stage by integrating it with maxtext in fsdp mode. We have plans for further optimizations to close the gap with pallas using different techniques such as:

  cancelled:
  Tests#12607:Pull request#2793synchronizebycopybara-service[bot]Bot
  test_834764107

  Dec 13, 20252h 9m 47s
  More actions for workflow run 12607

• Refactor dockerfiles and deprecate stable_stack and jax_ai_image mode

  cancelled:
  Tests#12606:Pull request#2822synchronizebySurbhiJainUSC
  refactor_docker

  Dec 13, 202511m 33s
  More actions for workflow run 12606

• Update llama2 normalization computation

  failed:
  Tests#12605:Pull request#2826openedbyNuojCheng
  chengnuojin-fix-llama2

  Dec 13, 202553m 38s
  More actions for workflow run 12605

• Refactor dockerfiles and deprecate stable_stack and jax_ai_image mode

  cancelled:
  Tests#12604:Pull request#2822synchronizebySurbhiJainUSC
  refactor_docker

  Dec 13, 202512m 10s
  More actions for workflow run 12604

• Add citation entry for repository

  failed:
  Tests#12603:Pull request#2825synchronizebyshuningjin
  cite-repo

  Dec 13, 202556m 56s
  More actions for workflow run 12603

• Add citation entry for repository

  cancelled:
  Tests#12602:Pull request#2825synchronizebyshuningjin
  cite-repo

  Dec 13, 20251m 7s
  More actions for workflow run 12602

• This change introduces a pure JAX implementation of flash attention to Maxtext, designed as a drop-in replacement for the existing Pallas kernel. In this cl we set up the stage by integrating it with maxtext in fsdp mode. We have plans for further optimizations to close the gap with pallas using different techniques such as:

  cancelled:
  Tests#12601:Pull request#2793synchronizebycopybara-service[bot]Bot
  test_834764107

  Dec 13, 202541m 11s
  More actions for workflow run 12601

• Fix broken links in docs

  completed successfully:
  Tests#12600:Pull request#2814synchronizebyhengtaoguo
  hengtaoguo-links

  Dec 13, 202537m 2s
  More actions for workflow run 12600

• Add citation entry for repository

  cancelled:
  Tests#12599:Pull request#2825openedbyshuningjin
  cite-repo

  Dec 13, 202511m 40s
  More actions for workflow run 12599

• Refactor dockerfiles and deprecate stable_stack and jax_ai_image mode

  cancelled:
  Tests#12598:Pull request#2822synchronizebySurbhiJainUSC
  refactor_docker

  Dec 13, 202521m 18s
  More actions for workflow run 12598

• Refactor dockerfiles and deprecate stable_stack and jax_ai_image mode

  cancelled:
  Tests#12597:Pull request#2822synchronizebySurbhiJainUSC
  refactor_docker

  Dec 13, 20258m 11s
  More actions for workflow run 12597

• This change introduces a pure JAX implementation of flash attention to Maxtext, designed as a drop-in replacement for the existing Pallas kernel. In this cl we set up the stage by integrating it with maxtext in fsdp mode. We have plans for further optimizations to close the gap with pallas using different techniques such as:

  cancelled:
  Tests#12596:Pull request#2793synchronizebycopybara-service[bot]Bot
  test_834764107

  Dec 13, 202523m 53s
  More actions for workflow run 12596

• Refactor dockerfiles and deprecate stable_stack and jax_ai_image mode

  cancelled:
  Tests#12595:Pull request#2822synchronizebySurbhiJainUSC
  refactor_docker

  Dec 13, 202511m 59s
  More actions for workflow run 12595

• This change introduces a pure JAX implementation of flash attention to Maxtext, designed as a drop-in replacement for the existing Pallas kernel. In this cl we set up the stage by integrating it with maxtext in fsdp mode. We have plans for further optimizations to close the gap with pallas using different techniques such as:

  cancelled:
  Tests#12594:Pull request#2793synchronizebycopybara-service[bot]Bot
  test_834764107

  Dec 13, 202512m 24s
  More actions for workflow run 12594