rawpair/python-nvidia-cuda repository overview

A Python machine learning container tuned for RawPair and heavy experimentation.

Features:

• Preinstalled: PyTorch, TensorFlow, Keras, Scikit-learn, XGBoost, LightGBM, CatBoost
• NLP/NLU stack: HuggingFace Transformers, Datasets, SentenceTransformers, FAISS
• Visualization tools: Matplotlib, Seaborn, Plotly
• Experiment tracking: Optuna, MLflow, Hydra

Best suited for prototyping, experimentation, and ML model development in RawPair with CUDA base image support.