-
Megatron-LM Public
Forked from NVIDIA/Megatron-LMOngoing research training transformer models at scale
Python Other UpdatedJun 26, 2024 -
-
-
-
xla Public
Forked from openxla/xlaA machine learning compiler for GPUs, CPUs, and ML accelerators
C++ Apache License 2.0 UpdatedNov 3, 2023 -
jax Public
Forked from jax-ml/jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python Apache License 2.0 UpdatedSep 27, 2023 -
JAX-Toolbox Public
Forked from NVIDIA/JAX-ToolboxJAX-Toolbox
Shell Apache License 2.0 UpdatedApr 25, 2023 -