Message Passing Neural Networks for Molecule Property Prediction
Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX
Multi-tool for semantic search
Python Stream Processing
Sacred is a tool to help you configure, andorganize IDSIA experiments
Automatically Visualize any dataset, any size
Trainable models and NN optimization tools
Your open-source LLM evaluation toolkit
Implementation of Imagen, Google's Text-to-Image Neural Network
Python framework for adversarial attacks, and data augmentation
Play couplet with seq2seq model
Standalone, small, language-neutral
We write your reusable computer vision tools
Hub of ready-to-use datasets for ML models
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Build cross-modal and multimodal applications on the cloud
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Scientific Visualisation Made Easy
PyTorch Lightning + Hydra. A very user-friendly template
Object detection architectures and models pretrained on the COCO data
CPU/GPU inference server for Hugging Face transformer models
A library of additional estimators and SageMaker tools based on scikit
Machine learning tool that allows you to train and test models
U-Net for RFI Detection based on @jakeret's implementation
Auto-diff neural network library for high-dimensional sparse tensors