Projects

DiffGuard is a small (67M parameters) enocder-only transformer fine-tuned on a large corpus of image/video generation prompt. It can classify prompts into two labels: Safe & NSFW (Not Safe For Work). It is aimed to be used as safety checker for text-to-image & text-to-video generative models. The model is Open-Source under the Apache 2.0 license. As of May 2025, the model has gathered more than 150K total downloads on HuggingFace since its release in June 2024... View More

ThreatDetect-C-Cpp is a derivative of ModernBERT-base trained to be a code classifier. Instead of binary classification ("safe", "unsafe"), The model classifies the input code into 7 labels: 'safe' (no vulnerability detected) and six other CWE weaknesses. ThreadDetect-C-Cpp can be integrated in code-related applications. For example, it can be used in pair with a code generator to detect vulnerabilities in the generated code. View More

GPUtopia is a an AI-based platform aimed to help AI Engineers/developpers/Data Scientist choose the best GPU usage strategy. The responses are grounded (via a modern RAG pipeline) by a relevant corpus of resources (e.g. The Ultrascale Playbook by HF). Just plug in your setup (number of GPUs, nodes, model size etc) and let the tool give you the best usage strategy (parallelism method etc). View More