Open-source library of optimized deep learning operations (matmul, convolution, attention) for CPUs (x64, AArch64, RISC-V) and Intel GPUs. Used by PyTorch, TensorFlow, OpenVINO, and ONNX Runtime.
- onednn
- oneapi
- performance
- cpp
47,133 repositories indexed
145–168 of 272 · Most stars
Open-source library of optimized deep learning operations (matmul, convolution, attention) for CPUs (x64, AArch64, RISC-V) and Intel GPUs. Used by PyTorch, TensorFlow, OpenVINO, and ONNX Runtime.
Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and models from research to production.
A highly efficient implementation of Gaussian Processes in PyTorch
A python library for self-supervised learning and latent world models on images.
AI Infra / AI Orchestration / AI Control Plane
OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.
Bayesian optimization in PyTorch
🛠 All-in-one web-based IDE specialized for machine learning and data science.
Fast, flexible and easy to use probabilistic modelling in Python.
Visual tracking library based on PyTorch.
PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.
Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2
Accelerated deep learning R&D
Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
Determined is an open-source machine learning platform that simplifies distributed training, hyperparameter tuning, experiment tracking, and resource management. Works with PyTorch and TensorFlow.
PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
Data manipulation and transformation for audio signal processing, powered by PyTorch
Multi-dimensional arrays (tensors) and numerical definitions for Elixir
Multilingual Automatic Speech Recognition with word-level timestamps and confidence
TranslatedCAPTCHA recognition
Automatic architecture search and hyperparameter optimization for PyTorch
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.
Reinforcement Learning in PyTorch
An open source library and framework for deep learning on satellite and aerial imagery.