Supercharge Your Model Training
Message Passing Neural Networks for Molecule Property Prediction
Hummingbird compiles trained ML models into tensor computation
DoWhy is a Python library for causal inference
A simple but complete full-attention transformer
An on-premises, OCR-free unstructured data extraction
A general fine-tuning kit geared toward image/video/audio diffusion
Superfast AI decision making and processing of multi-modal data
A Python package for segmenting geospatial data with the SAM
Python Package for ML-Based Heterogeneous Treatment Effects Estimation
Core ML tools contain supporting tools for Core ML model conversion
Flower: A Friendly Federated Learning Framework
Uniform Manifold Approximation and Projection
Deep and Machine Learning for Microscopy
Tensor Learning in Python
Open deep learning compiler stack for cpu, gpu, etc.
A machine learning library for detecting anomalies in signals
Detecting silent model failure. NannyML estimates performance
Tool for visualizing and tracking your machine learning experiments
A unified interface for distributed computing
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Jittor is a high-performance deep learning framework
A unified framework for scalable computing
Deep learning library
Experimental, AI/ML-powered and open sourced Marketing Mix Modeling