Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Mamba SSM architecture

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

Python Script to download hundreds of images from 'Google Images'. It is a ready-to-run code!

[ECCV 2024] codes of DiffBIR: Towards Blind Image Restoration with Generative Diffusion Prior

ROS Wrapper for Intel(R) RealSense(TM) Cameras

The PyTorch-based audio source separation toolkit for researchers

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

A library for easy and efficient manipulation of tensor networks.

A registry of publicly available datasets on AWS

Profiling and inspecting memory in pytorch

ALiPy: Active Learning in Python is an active learning python toolbox, which allows users to conveniently evaluate, compare and analyze the performance of active learning methods.

Autoencoder network for learning a continuous representation of molecular structures.

The official pytorch implemention of our ICML-2021 paper "SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Networks".

JAX implementation of the Llama 2 model

The Cone of Silence:

Various models for handling underexposure, overexposure, super-resolution, shadow removal, etc.

Complementary material for the submission RAL-ICRA 2017