Commits

Branch selector

 main

User selector

ruomingp

Datepicker

All time

Commit History

Commits on Oct 17, 2023

• Fixes large memory usage by pipeline (#128)

  Show description for 3e8117b
  ruomingp

  authored
  3e8117b

  Copy full SHA for 3e8117b

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

Commits on Oct 13, 2023

• 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

Commits on Oct 10, 2023

• 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 4, 2023

• Changes the API of `input_tf_data.tfds_dataset` (#91)

  Show description for 4ff105b
  ruomingp

  authored
  4ff105b

  Copy full SHA for 4ff105b

Commits on Oct 1, 2023

• Adds support for torch param conversion for LayerNormStateless. (#89)

  Show description for e7ef158
  ruomingp

  authored
  e7ef158

  Copy full SHA for e7ef158

Commits on Sep 28, 2023

• Makes Initializer a subclass of Configurable for consistency. (#85)

  Show description for 92b62a4
  ruomingp

  authored
  92b62a4

  Copy full SHA for 92b62a4

Commits on Sep 12, 2023

• Fixes `max(evaler_cfg.vlog, 3)`. (#69)

  ruomingp

  authored
  4c56c7a

  Copy full SHA for 4c56c7a

Commits on Sep 11, 2023

• Makes `ConfigBase.debug_string()` omit trivial default values. (#66)

  Show description for a91f080
  ruomingp

  authored
  a91f080

  Copy full SHA for a91f080

Commits on Sep 1, 2023

• Removes timestamp information in vprint (#51)

  Show description for e24fa8c
  ruomingp

  authored
  e24fa8c

  Copy full SHA for e24fa8c

Commits on Aug 28, 2023

• Adds read_state_spec to checkpointer.py. (#49)

  ruomingp

  authored
  b52fece

  Copy full SHA for b52fece

• Lists supported bundler types when reporting the unknown bundler type error (#50)

  ruomingp

  authored
  ae9805e

  Copy full SHA for ae9805e

Commits on Aug 27, 2023

• Adds Module.vprint, which uses jax.debug.print to print tensor contents (#48)

  Show description for 5c74c6c
  ruomingp

  authored
  5c74c6c

  Copy full SHA for 5c74c6c

Commits on Aug 6, 2023

• Provides a brief overview of AXLearn in README.md (#30)

  Show description for 4eacb5e
  ruomingp

  authored
  4eacb5e

  Copy full SHA for 4eacb5e