Splinter0/tensorflow-rce
RCE PoC for Tensorflow using a malicious Lambda layer
This project helps security researchers understand how a malicious remote code execution payload can be embedded within a TensorFlow machine learning model. It demonstrates how an attacker could craft a model file (an .h5 file) to include a reverse shell, and how that model could be used to compromise a system when loaded. Security professionals and penetration testers working with machine learning systems would use this to investigate potential vulnerabilities.
No commits in the last 6 months.
Use this if you are a security researcher or penetration tester looking to understand and demonstrate remote code execution vulnerabilities in TensorFlow models.
Not ideal if you are a data scientist or machine learning engineer looking for tools to build or deploy secure models, as this is an exploit demonstration, not a defensive tool.
Stars
61
Forks
8
Language
Python
License
MIT
Category
Last pushed
Jul 30, 2025
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/Splinter0/tensorflow-rce"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
TalEliyahu/Awesome-AI-Security
Curated resources, research, and tools for securing AI systems
The-Art-of-Hacking/h4cker
This repository is maintained by Omar Santos (@santosomar) and includes thousands of resources...
aw-junaid/Hacking-Tools
This Repository is a collection of different ethical hacking tools and malware's for penetration...
jiep/offensive-ai-compilation
A curated list of useful resources that cover Offensive AI.
Kim-Hammar/csle
A research platform to develop automated security policies using quantitative methods, e.g.,...