fenghuizhang/torchrec
Pytorch domain library for recommendation systems
Repositories
fenghuizhang/vllm
A high-throughput and memory-efficient inference and serving engine for LLMs
fenghuizhang/pt_xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
fenghuizhang/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
fenghuizhang/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more