All pull requests

Pull requests

• [DOC] Specify cupy versions according to CUDA versions

  #951·serach24 opened Aug 20, 2023·

  ·

• task 801 - launch physical meshes after compilation

  #938·haifaksh opened Jun 18, 2023·

  ·
  11

• Rebase ALPA to JAX 0.4.10

  #926·yhtang opened May 12, 2023·5 tasks·

  ·
  1

• [EXAMPLE] Add llama finetune

  #923·ZYHowell opened Apr 29, 2023Collaborator·

  ·

• feat(compiler): implement last-seen heuristic for resharding source

  #917·jon-chuang opened Apr 8, 2023·

  ·

• Use shell script to drive jaxlib build

  #909·yhtang opened Mar 30, 2023·

  ·
  5

• Create devcontainer.json

  #899·Spina7demon opened Mar 18, 2023·

  ·

• [WIP] OPT training improvements

  #858·zhisbug opened Jan 17, 2023Member·

  ·

• [WIP] [FEATURE] Improve layer construction for inference

  #800·zhuohan123 opened Dec 4, 2022Member·

  ·
  2