Tensors and Dynamic neural networks in Python with strong GPU acceleration
- neural-network
- autograd
- gpu
- numpy
Repository Index
64,764 visible repositories in the index
1–24 of 66 · Most stars
Simulation platform for general-purpose robotics & embodied AI learning.
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
AI Toolkit for Healthcare Imaging
- healthcare-imaging
- deep-learning
- medical-image-computing
- medical-image-processing
Efficiently computes derivatives of NumPy code.
- autograd
- automatic-differentiation
- numpy
- python
A Python framework for GPU-accelerated simulation, robotics, and machine learning.
- differentiable-programming
- python
- gpu-acceleration
- nvidia
Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, Materials Science and Biology
- drug-discovery
- quantum-chemistry
- deep-learning
- biology
Physics Informed Deep Learning: Data-driven Solutions and Discovery of Nonlinear Partial Differential Equations
A PyTorch Library for Accelerating 3D Deep Learning Research
- pytorch
- 3d-deep-learning
- neural-networks
- differentiable-rendering
A library for scientific machine learning and physics-informed learning
- neural-network
- deep-learning
- scientific-machine-learning
- pinn
DeepVariant is an analysis pipeline that uses a deep neural network to call genetic variants from next-generation DNA sequencing data.
- tensorflow
- deep-neural-network
- genomics
- science
MNE: Magnetoencephalography (MEG) and Electroencephalography (EEG) in Python
- python
- neuroscience
- electroencephalography
- magnetoencephalography
PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.
A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
- rl
- reinforcement-learning
- stable-baselines
- openai
Implementation of papers in 100 lines of code.
使用机器学习算法完成对12306验证码的自动识别
- deep-learning
- mathematics
- captcha
- 12306
POT : Python Optimal Transport
- optimal-transport
- numerical-optimization
- machine-learning
- emd
Message Passing Neural Networks for Molecule Property Prediction
- machine-learning
- chemistry
- neural-networks
- drug-discovery
Optax is a gradient processing and optimization library for JAX.
This package contains deep learning models and related scripts for RoseTTAFold
Source code of PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).
- python
- genetic-algorithm
- optimization
- numpy
Deep probabilistic analysis of single-cell and spatial omics data
- scrna-seq
- variational-bayes
- variational-autoencoder
- cite-seq
Implementation of DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking
- diffusion-models
- docking
- binding
- computational-biology