All pull requests

Pull requests

• feat(evaluation): add buffer donation support for attention kernels

  #107·YuxuanChen opened Sep 24, 2026Collaborator·

  ·
  3

• Add JaxBench level 2

  #102·NinaCai opened Sep 9, 2026Collaborator·

  ·
  18

• Add 8 priority human optimized kernels and SWA into JaxBench level 2

  #101·shangkunwang01 opened Sep 9, 2026Collaborator·

  ·
  23

• Add rpav3, gmmv2, causal_conv1d_gdn to JaxBench

  #100·cccathygao opened Sep 4, 2026Collaborator·

  ·
  6

  

• Split monolithic coworker.py into modular CLIs and add root Makefile

  #92·prksandeep opened Aug 17, 2026Collaborator·

  ·
  6

• Move MaxKernel to MaxKernel/adk

  #89·cccathygao opened Aug 14, 2026Collaborator·

  ·
  1

• beam_search: parallel auto_agent workers in a beam_search orchestrator loop

  #59·ligh-svg opened Jun 25, 2026Collaborator·

  ·
  6

• refactor(auto_agent): extract server schemas, isolate mock server, and add fake evaluator

  #58·ligh-svg opened Jun 25, 2026Collaborator·

  ·

• auto_agent: Pre-requisites of auto_agent beam search prototype using mock backends

  #57·ligh-svg opened Jun 25, 2026Collaborator·

  ·

• JAXBench: Add back jaxbench profiling and block size tuning scripts

  #52·charleshong3 opened Jun 16, 2026Contributor·

  ·
  1

• JAXBench: Fix Megablox GMM no-op exploit: use realistic non-underflowing inputs

  #49·charleshong3 opened Jun 10, 2026Contributor·

  ·
  1

• chore: update default WORKDIR to include an artifacts subdirectory in prepare_hitl_agent.sh

  #32·shangkunwang01 opened Apr 22, 2026Collaborator·

  ·

• MaxCode: Add REST API, update MCP server, and refresh documentation

  #30·gvanica opened Apr 22, 2026Collaborator·

  ·
  1

• MaxCode: Add MaxText conversion path with dedicated agent and demo scripts

  #29·gvanica opened Apr 22, 2026Collaborator·

  ·
  1

• MaxCode: Add demo pipeline scripts for end-to-end conversion workflow

  #28·gvanica opened Apr 22, 2026Collaborator·

  ·
  1

• MaxCode: Integrate MergeAgent into PrimaryAgent pipeline and update conversion tools

  #27·gvanica opened Apr 22, 2026Collaborator·

  ·
  1

• MaxCode: Overhaul migration prompts with JAX best practices and MaxText support

  #26·gvanica opened Apr 22, 2026Collaborator·

  ·
  1

• MaxCode: Add VerificationAgent and ValidationAgent for conversion quality scoring

  #25·gvanica opened Apr 22, 2026Collaborator·

  ·
  1

• MaxCode: Add MergeAgent for file discovery, import-graph analysis and merging

  #24·gvanica opened Apr 22, 2026Collaborator·

  ·
  1

• MaxCode: Add RAG knowledge base — JAX/Flax docs and targeted migration rules

  #23·gvanica opened Apr 22, 2026Collaborator·

  ·
  1

• Pr/agent layer integration

  #17·gvanica opened Apr 15, 2026Collaborator·

  ·