Jia Chenwei's repositories
archai
Accelerate your Neural Architecture Search (NAS) through fast, reproducible and modular research.
awesome-rust
A curated list of Rust code and resources.
alpa
Training and serving large-scale neural networks with auto parallelization.
complete-javascript-course
Starter files, final projects, and FAQ for my Complete JavaScript course
efficient-kan
An efficient pure-PyTorch implementation of Kolmogorov-Arnold Network (KAN).
forecast-mae
[ICCV'2023] Forecast-MAE: Self-supervised Pre-training for Motion Forecasting with Masked Autoencoders
GameFormer
[ICCV 2023 Oral] Game-theoretic modeling and learning of Transformer-based interactive prediction and planning
gpt-2
Code for the paper "Language Models are Unsupervised Multitask Learners"
grouped-query-attention-pytorch
(Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints" (https://arxiv.org/pdf/2305.13245.pdf)
lsq-net
Unofficial implementation of LSQ-Net, a neural network quantization framework
mistral-inference
Official inference library for Mistral models
MIT-6.5940
All Homeworks for TinyML and Efficient Deep Learning Computing 6.5940 • Fall • 2023 • https://efficientml.ai
multidim-positional-encoding
An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
onnx-simplifier
Simplify your onnx model
openwrt-package
openwrt插件库-18.06-21.02-22.03插件
pykan
Kolmogorov Arnold Networks
Reinforcement_Learning
Reinforcement learning tutorials
ros2_rust
Rust bindings for ROS2
rustlings
:crab: Small exercises to get you used to reading and writing Rust code!
SimVP-Simpler-yet-Better-Video-Prediction
The official implementation of the CVPR'2022 paper SimVP: Simpler Yet Better Video Prediction.
TNT-Trajectory-Prediction
A Pytorch Implementation of TNT: Target-driveN Trajectory Prediction
tvm-riscv-demo
Demonstrator on running TVM on RISC-V with RN18 and KWS examples
wanda
A simple and effective LLM pruning approach.
YOLO-World
[CVPR 2024] Real-Time Open-Vocabulary Object Detection
yolov9
Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information