Ma Wenhui's repositories
apple_amx
GEMM with apple silicon AMX registers VS A64 NEON
DeepCoMP
Dynamic multi-cell selection for cooperative multipoint (CoMP) using (multi-agent) deep reinforcement learning
examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
json.lua
A lightweight JSON library for Lua
ppl.nn
A primitive library for neural network
PRIMAL2
Training code PRIMAL2 - Public Repo
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
ray
An open source framework that provides a simple, universal API for building distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library.
rmf_ros2
Internal ROS infrastructure for RMF
tutorials
PyTorch tutorials.