All Tools
J
Dev ToolsFreeOpen Source
JITTOR
JIT deep learning framework with meta-operators
Apache-2.0
ABOUT
Training custom models often needs kernels specialized per graph, not a fixed op set. Jittor JIT-compiles meta-operators into CUDA and C++ so vision, geometry, and generation networks train with a Python Module API without hand-writing those kernels.
INSTALL
pip install jittorINTEGRATION GUIDE
1. Train image recognition and detection models with a PyTorch-like API
2. JIT-compile custom convolution or geometry operators for CUDA
3. Port research models that need unified dynamic and static graphs
4. Run Jittor model libraries for segmentation, generation, or rendering
TAGS
pythondeep-learningjitcudapytorch-liketsinghuatraining