jklaise
Follow
ML tools
Extension for Scikit-learn is a seamless way to speed up your Scikit-learn application
Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.
skops is a Python library helping you share your scikit-learn based models and put them in production
A Bayesian optimization toolbox built on TensorFlow
Diffusers-Interpret 🤗🧨🕵️♀️: Model explainability for 🤗 Diffusers. Get explanations for your generated images.
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.
Feathr – A scalable, unified data and AI engineering platform for enterprise
Flax is a neural network library for JAX that is designed for flexibility.
Annotating tensor shapes using Python types
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.
KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
Milvus is a high-performance, cloud-native vector database built for scalable vector ANN search
Parallel Hyperparameter Tuning in Python
The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.
A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different angles at the click of a button.