LoreonLabsPlatform
DocsHome
  • Overview

Intelligence

  • Markets
  • Builders
  • Research
  • Ecosystems
  • Launchpads
  • Search
Ecosystems

Python

paxml

khshah6/paxml

PythonEmerging
GitHub
Stars
—
Forks
—
Contributors
8
Last push
42mo ago

Recent commits

Latest commits.

  • document the caveat of setting noise_multiplier when training with pmap
    cd26705Chiyuan Zhang42mo ago
  • Add instrumentation to track different types of pax jobs.
    ea2a11cpax authors42mo ago
  • Enable auto-sharding for evaluate().
    17e204eGuangda Lai42mo ago
  • exclude late-evaluated tuning knobs from eval and decoder dataset.
    2f91d52Daiyi Peng42mo ago
  • Report training summaries using `step+1`.
    3eff59eLaurent El Shafey42mo ago
Remove _global_avals from infer_params because everything is global in pjit after jax.Array was enabled.
b4438bbYash Katariya42mo ago
  • Microbatch and Augmentation Multiplicity DP-SGD support in sgf
    95610e1Chiyuan Zhang42mo ago
  • Updated the logic that increments or synchronizes step_i such that the train
    827b6a9Sertan Girgin42mo ago

    • Top contributors

    Builders behind this project.

    laurentes
    88 commits
    jysohn23
    55 commits
    cpgaffney1
    40 commits
    daiyip
    36 commits
    aaroey
    34 commits
    edloper
    17 commits
    zhangyujing
    16 commits
    zhangqiaorjc
    15 commits