Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 25, 2023

• Update README.md

  CrypticRevenger

  authored
  79f1ede

  Copy full SHA for 79f1ede

• Check for 'WAITING_FOR_RESOURCES' state. (#144)

  

  Mark Lee

  authored
  8c6866b

  Copy full SHA for 8c6866b

Commits on Oct 24, 2023

• Add env-vars to launch which improve checkpointing reliability. (#141)

  tgunter

  authored
  f1b05a1

  Copy full SHA for f1b05a1

Commits on Oct 23, 2023

• Add support for blob-storage external deps in axlearn.cloud.common.bundler. (#136)

  Show description for bdb6066
  tgunter

  authored
  bdb6066

  Copy full SHA for bdb6066

Commits on Oct 22, 2023

• Replaces config_for_partial_function with config_for_function. (#137)

  

  Mark Lee

  authored
  3808aa4

  Copy full SHA for 3808aa4

• Implement rank_by_value with support for tied ranks. (#135)

  Show description for 8b3b7f3
  tuzhucheng

  authored
  8b3b7f3

  Copy full SHA for 8b3b7f3

Commits on Oct 20, 2023

• Infers mesh shape when using state builder. (#133)

  

  Mark Lee

  authored
  ce0c453

  Copy full SHA for ce0c453

• fix rope maxlen (#134)

  Show description for d271506
  

  alex8937

  and

  Chen Chen

  authored
  d271506

  Copy full SHA for d271506

• Add query_scale and key_scale to MultiheadAttention. (#122)

  Show description for 162f1e8
  apghml

  authored
  162f1e8

  Copy full SHA for 162f1e8

Commits on Oct 19, 2023

• Keep user-specified find_links in tpu_runner.py (#132)

  ya5ut

  authored
  4a77cd8

  Copy full SHA for 4a77cd8

• Adds audio frontend utils. (#131)

  

  Mark Lee

  authored
  c9d3189

  Copy full SHA for c9d3189

Commits on Oct 18, 2023

• Add TPU flash-attention implementation. (#129)

  Show description for 6789f84
  tgunter

  authored
  6789f84

  Copy full SHA for 6789f84

Commits on Oct 17, 2023

• Fixes large memory usage by pipeline (#128)

  Show description for 3e8117b
  ruomingp

  authored
  3e8117b

  Copy full SHA for 3e8117b

• Makes maybe_set_config more flexible. (#127)

  

  Mark Lee

  authored
  13d3518

  Copy full SHA for 13d3518

Commits on Oct 16, 2023

• Similar to a previous change to repeat.py, changes pipeline.py to split stacked summary values. (#125)

  Show description for faf279b
  ruomingp

  authored
  faf279b

  Copy full SHA for faf279b

Commits on Oct 15, 2023

• Updates chex version to >= 0.1.7 to be compatible with jax 0.4.18. (#124)

  ruomingp

  authored
  e6b978f

  Copy full SHA for e6b978f

• fix typo: traier -> trainer (#120)

  malivinayak

  authored
  1194631

  Copy full SHA for 1194631

Commits on Oct 14, 2023

• Exclude arm images for booting VM. (#123)

  

  Mark Lee

  authored
  282e5ed

  Copy full SHA for 282e5ed

• Bump jax/jaxlib to 0.4.18. (#119)

  tgunter

  authored
  200c791

  Copy full SHA for 200c791

Commits on Oct 13, 2023

• make theta configurable in RoPE (#116)

  Show description for 627196a
  

  alex8937

  and

  Chen Chen

  authored
  627196a

  Copy full SHA for 627196a

• Adds a pointer to https://github.com/apple/axlearn/blob/main/docs/01-start.md#preparing-the-cli in the error message. (#114)

  Show description for b283c09
  ruomingp

  authored
  b283c09

  Copy full SHA for b283c09

• update (#118)

  Show description for ccd172e
  

  zbwglory

  and

  bowen-zhang9

  authored
  ccd172e

  Copy full SHA for ccd172e

Commits on Oct 12, 2023

• Add Nested type. (#117)

  apghml

  authored
  e3d9f46

  Copy full SHA for e3d9f46

• Add support to infer -1 in mesh shape (#113)

  Show description for 4693a0e
  lyttonhao

  authored
  4693a0e

  Copy full SHA for 4693a0e

Commits on Oct 11, 2023

• Add option of whether to apply rotary position embeddings on value (#111)

  Show description for 4c07bdc
  

  alex8937

  and

  Chen Chen

  authored
  4c07bdc

  Copy full SHA for 4c07bdc

• Tighten cryptography/pyopenssl pins. (#112)

  

  Mark Lee

  authored
  e8a2131

  Copy full SHA for e8a2131

• fix RoPE dtype not equal float32 (#110)

  Show description for 40c8722
  

  alex8937

  and

  Chen Chen

  authored
  40c8722

  Copy full SHA for 40c8722

• Make repeat in RepeatedTransformerLayer configurable. (#109)

  Show description for fddf05a
  

  xianzhidu

  and

  Xianzhi Du

  authored
  fddf05a

  Copy full SHA for fddf05a

• Decouples bastion from GCP. (#108)

  

  Mark Lee

  authored
  fff901e

  Copy full SHA for fff901e

• Fix typo in imagenet_adversarial_text/README.md (#59)

  Show description for 7d1f3e8
  eltociear

  authored
  7d1f3e8

  Copy full SHA for 7d1f3e8

Commits on Oct 10, 2023

• Workaround JAX PJIT bug that was breaking image summaries. (#107)

  apghml

  authored
  846700a

  Copy full SHA for 846700a

• [1/N] Adding Rotary position embedding test against LLAMA implementation (#105)

  Show description for 1c230ba
  

  zbwglory

  and

  bowen-zhang9

  authored
  1c230ba

  Copy full SHA for 1c230ba

• Splits output summaries of repeat layers (#106)

  Show description for 1b034af
  ruomingp

  authored
  1b034af

  Copy full SHA for 1b034af

Commits on Oct 8, 2023

• Adds `SpmdTrainer.Config.mesh_rules` and `--mesh_selector` to override trainer mesh configuration. (#104)

  Show description for a3a63d2
  ruomingp

  authored
  a3a63d2

  Copy full SHA for a3a63d2

Commits on Oct 7, 2023

• Inherit GCP pins in TPU extras. (#103)

  

  Mark Lee

  authored
  bcacabb

  Copy full SHA for bcacabb

Pagination

Previous

Next