100 dependents
Package Description Downloads/month
google clu
CLU lets you write beautiful training loops in JAX. 830K
Easy to use stem (e.g. instrumental/vocals) separation from CLI or as a python p... 364K
A Python library for extracting structured information from unstructured text us... 214K
OpenSpiel is a collection of environments and algorithms for research in general... 172K
Differentiable, Hardware Accelerated, Molecular Dynamics 95K
Deep probabilistic analysis of single-cell and spatial omics data 77K
Open source code for AlphaFold. 73K
Massively parallel rigidbody physics simulation on accelerator hardware. 53K
(EasyDel Former) is a utility library designed to simplify and enhance the devel... 24K
Modular, scalable library to train ML models 24K
The Learning Interpretability Tool: Interactively analyze ML models to understan... 21K
Toward High-Accuracy Open-Source Biomolecular Structure Prediction. 13K
12K
Easy to use audio stem separation with a UI, using various models from UVR train... 12K
Inference-only Band-Split Roformer separation toolkit 10K
Biomolecular emulator 6K
Nucleobench optimizers and tasks. 5K
TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow pla... 4K
(EasyDel Former) is a utility library designed to simplify and enhance the devel... 4K
Pipeline allows massive screening using alphafold 3K
Use bacpipe to streamline the process of generating embeddings and analysing you... 3K
A fully open source biomolecular structure prediction model based on AlphaFold3 2K
Unlearning Algorithms 2K
Tooling for agile modeling on large machine perception embedding databases. 2K
Sandbox for Computational Protein Design 2K
Open-Source Frontier Voice AI 2K
PXMeter is a comprehensive toolkit for evaluating the quality of structu... 2K
A modular library for visual 4D scene understanding 2K
Scalable drone simulation using JAX. 2K
Score-based Diffusion models in JAX. 1K
Repository of model architectures and network weights used for CellMap segmentat... 1K
AIDO.ModelGenerator is a software stack powering the development of an AI-driven... 1K
Fast, lightweight and parallelised simulation-based inference in JAX. 894
🌾 OAT: A research-friendly framework for LLM online alignment, including reinfor... 788
Remove background music from videos - for accessibility and personal use 686
A Framework for Reinforcement Learning in Games (forked from OpenSpiel) 677
Lightweight antibody structure prediction tool (arXiv:2507.09054) 665
LLM-powered structured information extraction from unstructured text, with sourc... 655
The CLRS Algorithmic Reasoning Benchmark. 645
Hyperparameter search and metric visualization tool for personal research. 561
Inference-only Mel-Band Roformer vocal separation toolkit 542
Implementation of Transformer Flows (Apple ML) in JAX and Equinox. 509
Towards Embodied AI with MuscleMimic: Unlocking full-body musculoskeletal motor ... 485
Implementation of Rectified Flows in JAX and Equinox. 468
Felafax is building AI infra for non-NVIDIA GPUs 451
a pip installable version of LigandMPNN with pre-trained models included 430
A small example package 400
IntelliFold: A Controllable Foundation Model for General and Specialized Biomole... 394
OneScience是基于先进的深度学习框架打造的科学计算工具包,旨在通过一系列高度集成的组件加速科学研究和技术开发进程。 388
LXM3: XManager launch backend for HPC clusters 387