Neural Network Compression Framework for enhanced OpenVINO
An interactive NVIDIA-GPU process viewer and beyond
A python library for easy manipulation and forecasting of time series
Scalable and user friendly neural forecasting algorithms.
Medical imaging toolkit for deep learning
End-to-End Library for Continual Learning based on PyTorch
TFDS is a collection of datasets ready to use with TensorFlow,
A library for scientific machine learning & physics-informed learning
Spatiotemporal Signal Processing with Neural Machine Learning Models
The toolkit to test, validate, and evaluate your models and surface
BioNeMo Framework: For building and adapting AI models
Everything you need to build state-of-the-art foundation models
Data processing for and with foundation models
Real-Time High-Resolution Background Matting
Making large AI models cheaper, faster and more accessible
Simulation of spiking neural networks (SNNs) using PyTorch
Algorithms for explaining machine learning models
Tensor Learning in Python
The easiest way to use deep metric learning in your application
Solve end to end problems using Llama model family
An Autonomous LLM Agent for Complex Task Solving
Optax is a gradient processing and optimization library for JAX
SkyPilot: Run AI and batch jobs on any infra
Python library for Representation Learning on Knowledge Graphs
Experiment tracking and metric logging for Amazon SageMaker notebooks