Roadmap

[yijunyu]

Tasks

Extract Learning Samples

• Extract functions from Crates-IO index and split them into safe and unsafe classes
• Extract blocks from Crates-IO index and split them into safe and unsafe classes
• Annotate extracted code with inlay hints

Train a model

• Use CodeBERT to train a classification model on safe/unsafe functions
• Use CodeBERT to train a classification model on safe/unsafe blocks
• Convert the PyTorch model into an Rust_BERT model on Huggingface for functions
• Convert the PyTorch model into an Rust_BERT model on Huggingface for blocks

Inference with the model

• Develop a VSCode Plugin to Classify a function at cursor
• Compare the performance between Rust_BERT and CodeBERT for inferencing time
• Develop a VSCode Plugin to Classify a block at cursor