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 112 · Most stars
Google Research
Simulation platform for general-purpose robotics & embodied AI learning.
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
- autodiff
- deep-learning
- machine-learning
- rust
Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...
- java
- gpu
- deeplearning
- neural-nets
Open3D: A Modern Library for 3D Data Processing
- mesh-processing
- computer-graphics
- opengl
- cpp
CUDA Templates and Python DSLs for High-Performance Linear Algebra
- cuda
- deep-learning
- deep-learning-library
- cpp
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 Cloud Native Batch System (Project under CNCF)
- batch-systems
- kubernetes
- golang
- hpc
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.
Massively parallel rigidbody physics simulation on accelerator hardware.
Multi-language suite for high-performance solvers of differential equations and scientific machine learning (SciML) components. Ordinary differential equations (ODEs), stochastic differential equations (SDEs), delay differential equations (DDEs), differential-algebraic equations (DAEs), and more in Julia.
- differential-equations
- differentialequations
- julia
- ode
Highly cited and useful papers related to machine learning, deep learning, AI, game theory, reinforcement learning
- deep-learning
- machine-learning
- machine-learning-algorithms
- artificial-intelligence
A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
- rl
- reinforcement-learning
- stable-baselines
- openai