104 dependents
Package Description Downloads/month
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ... 1.6M
Probabilistic reasoning and statistical analysis in TensorFlow 1.1M
A Python interface for Reinforcement Learning environments. 417K
Google DeepMind's software stack for physics-based simulation and Reinforcement ... 408K
Probabilistic reasoning and statistical analysis in TensorFlow 386K
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ... 290K
A toolkit to optimize ML models for deployment for Keras and TensorFlow, includi... 105K
Open source code for AlphaFold. 73K
Boltz 63K
Reverb is an efficient and easy-to-use data storage and transport system designe... 54K
A multi-backend implementation of the Keras API, with support for TensorFlow, JA... 44K
ORB forcefield models from Orbital Materials 40K
An open-source framework for machine learning and other computations on decentra... 33K
Launchpad is a library that simplifies writing distributed programs and seamless... 27K
Google's differential privacy libraries. 23K
18K
Library for training machine learning models with privacy for training data 16K
the Amesa core functionality required to apply the Machine Teaching paradigm 14K
AutoML library for deep learning 12K
A suite of tools that users, both novice and advanced can use to optimize machin... 10K
the Composabl core functionality required to apply the Machine Teaching paradigm 8K
Reverb is an efficient and easy-to-use data storage and transport system designe... 8K
A Python utility for wrapping Rosetta command line tools. 7K
Runs openpi policy interface by websocket connection to a server. 7K
Biomolecular emulator 6K
TF2JAX: Convert TensorFlow to JAX 6K
Shared utilities and training infrastructure for biomolecular structure predicti... 6K
A GPU-accelerated library containing highly optimized building blocks and an exe... 5K
A high-performance, memory-efficient inference server for diffusion models, comp... 5K
Reinforcement learning algorithms in RLlib 4K
NVIDIA DALI nightly for CUDA 11.0. Git SHA: 2be08c56f2be9ec8055256256039eb534ab... 4K
Boltz for VSYNTHES 3K
Pipeline allows massive screening using alphafold 3K
A GPU-accelerated library containing highly optimized building blocks and an exe... 3K
A Jax package for approximate curvature estimation and optimization using KFAC. 3K
the Amesa core functionality required to apply the Machine Teaching paradigm 2K
Launchpad is a library that simplifies writing distributed programs and seamless... 2K
A client for interacting with endpoints of the FutureHouse crow service. 2K
2K
trfl is a library of building blocks for reinforcement learning algorithms. 2K
Sandbox for Computational Protein Design 2K
Dash + Leaflet = 🥳 2K
A simple and efficient Minecraft Agent development kit for AI research. 2K
Continuous-discrete dynamical systems with JAX and related libraries. 1K
High-performance open-source deep-learning framework for exploring, building and... 1K
juxtapose: Multi-Person Pose Tracking Inference SDK with RTMDet, YOLOv8, GDino, ... 1K
Convert a Numeric PDDL domain into an Gymnasium environment 1K
AIDO.ModelGenerator is a software stack powering the development of an AI-driven... 1K
Python drug discovery toolkit for Boltz2 structure and affinity prediction, Bolt... 992
Federated Learning for Health 984