hx89's repositories
FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Language:C++NOASSERTION000
paxml
Pax 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 leading model flop utilization rates.
Language:PythonApache-2.0000
Language:PythonApache-2.0000
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:C++NOASSERTION000
tutorials
PyTorch tutorials.
Language:Jupyter NotebookBSD-3-Clause000