-
maxtext Public
Forked from AI-Hypercomputer/maxtextA simple, performant and scalable Jax LLM!
Python Apache License 2.0 UpdatedDec 3, 2025 -
maxdiffusion Public
Forked from AI-Hypercomputer/maxdiffusionPython Apache License 2.0 UpdatedOct 29, 2025 -
-
xla Public
Forked from openxla/xlaA machine learning compiler for GPUs, CPUs, and ML accelerators
C++ Apache License 2.0 UpdatedMay 29, 2025 -
cutlass Public
Forked from NVIDIA/cutlassCUDA Templates for Linear Algebra Subroutines
C++ Other UpdatedMar 12, 2025 -
-
paxml Public
Forked from google/paxmlPax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…
Python Apache License 2.0 UpdatedJun 10, 2024 -
pytorch Public
Forked from pytorch/pytorchTensors and Dynamic neural networks in Python with strong GPU acceleration
C++ Other UpdatedMar 4, 2022 -
FBGEMM Public
Forked from pytorch/FBGEMMFB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
C++ Other UpdatedJan 28, 2022 -
tutorials Public
Forked from pytorch/tutorialsPyTorch tutorials.
Jupyter Notebook BSD 3-Clause "New" or "Revised" License UpdatedJul 28, 2020