Stars
A toolkit for developing and comparing reinforcement learning algorithms.
Kedro is a toolbox for production-ready data science. It uses software engineering best practices to help you create data engineering and data science pipelines that are reproducible, maintainable,…
Code for the paper "Jukebox: A Generative Model for Music"
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
A lightweight yet powerful audio-to-MIDI converter with pitch bend detection
TorchCFM: a Conditional Flow Matching library
a generalist algorithm for cellular segmentation with human-in-the-loop capabilities
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
General code to convert a trained keras model into an inference tensorflow model
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Code for the paper "Planning with Diffusion for Flexible Behavior Synthesis"
StarDist - Object Detection with Star-convex Shapes
NequIP is a code for building E(3)-equivariant interatomic potentials
A generative model for programmable protein design
An open library for the analysis of molecular dynamics trajectories
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
Optimizing AlphaFold Training and Inference on GPU Clusters
Allegro is an open-source code for building highly scalable and accurate equivariant deep learning interatomic potentials
Converting real-time EEG into sounds, music and visual effects
Framework for benchmarking vector search engines
Set of tools to interact with LTSpice. See README file for more information.
Typing-toolbox for Python 3 _and_ 2.7 w.r.t. PEP 484.
Python interface for the RCSB PDB search API.
A Python library for dynamic dispatch based on module versions and backends.
A tiny library for stochastic dataset caching in PyTorch.