JAX-Toolbox/.github/container/manifest.yaml at 078d865cc42a3bca60b9a01c30cc3a6b085dad94 · NVIDIA/JAX-Toolbox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
jax:
  url: https://github.com/google/jax.git
  tracking_ref: main
  latest_verified_commit: 3f6b59a67701a6047ff14c4e630943453ba3f491
  mode: git-clone
xla:
  url: https://github.com/openxla/xla.git
  tracking_ref: main
  latest_verified_commit: b3a188cb583517866d6996833dd3faf341b7388e
  mode: git-clone
flax:
  url: https://github.com/google/flax.git
  mirror_url: https://github.com/nvjax-svc-0/flax.git
  tracking_ref: main
  mode: git-clone
  patches:
    pull/3340/head: file://patches/flax/PR-3340.patch # Add Sharding Annotations to Flax Modules
  latest_verified_commit: a31129e2f8a2f5ca6ccc8492e536d2b28359f747
transformer-engine:
  url: https://github.com/NVIDIA/TransformerEngine.git
  tracking_ref: main
  latest_verified_commit: 68493d2d55ac37e540301467b278bdb1c2019e81
  mode: git-clone
pydantic:
  version: X.Y.Z
  mode: pip-constraint
dllogger:
  url: https://github.com/NVIDIA/dllogger.git
  tracking_ref: master
  latest_verified_commit: 0478734ff7be75adde8d160e04872664d1c62e5f
  mode: pip-vcs
jestimator:
  url: https://github.com/google-research/jestimator.git
  tracking_ref: main
  latest_verified_commit: 326824a477feef6d92e6f44175000907a72899ca
  mode: pip-vcs
optax:
  url: https://github.com/google-deepmind/optax.git
  tracking_ref: main
  latest_verified_commit: 9ed47fd2889030984f38e286aa6c3e269243dd4d
  mode: pip-vcs
seqio:
  url: https://github.com/google/seqio.git
  tracking_ref: main
  latest_verified_commit: aaa0e111c5e9d8fb49bfa992f2272a121a95f741
  mode: pip-vcs
maxtext:
  url: https://github.com/google/maxtext.git
  tracking_ref: main
  latest_verified_commit: 87e6bdd419d98780b22e54a05049e172fb07ab39
  mode: git-clone
mlperf-logging:
  url: https://github.com/mlcommons/logging.git
  tracking_ref: master
  latest_verified_commit: 7162c29326645231a27357c0c345997207c486c3
  mode: pip-vcs
equinox:
  url: https://github.com/patrick-kidger/equinox.git
  tracking_ref: main
  latest_verified_commit: 9336d820b2ef66b360fc6cb3611b3da5cefbb14d
  mode: git-clone
orbax-checkpoint:
  url: https://github.com/google/orbax.git
  tracking_ref: main
  latest_verified_commit: 79ee387ae63933924d04a7fd7e76a2ffc9041972
  mode: pip-vcs
pathwaysutils:
  url: https://github.com/google/pathways-utils.git
  tracking_ref: main
  latest_verified_commit: cf82422144fc28f580abab66422f3a4668da96cb
  mode: pip-vcs
axlearn:
  url: https://github.com/apple/axlearn.git
  mirror_url: https://github.com/nvjax-svc-0/axlearn.git
  tracking_ref: main
  mode: git-clone
  patches: {}
  latest_verified_commit: 34aed43c1b6e581c4753d06d7cbeedd5010ef26b
qwix:
  url: https://github.com/google/qwix.git
  tracking_ref: main
  latest_verified_commit: 763e19ff07daa89ba7290ef67dc05fbada95f939
  mode: git-clone
tunix:
  url: https://github.com/google/tunix.git
  tracking_ref: main
  latest_verified_commit: ddf95ff43f35eae7dd41aaacb4ab5cf2c12e2d48
  mode: git-clone
torchax:
  url: https://github.com/google/torchax.git
  tracking_ref: main
  latest_verified_commit: ac59e3d2becf079b336a8caa831746bb0326e6d4
  mode: git-clone
alphafold:
  url: https://github.com/google-deepmind/alphafold3.git
  tracking_ref: main
  latest_verified_commit: 7ecd64ff0e3df58bfd1cdca62e6064b64c1e3a21
  mode: git-clone