Tensors and Dynamic neural networks in Python with strong GPU acceleration
- neural-network
- autograd
- gpu
- numpy
Repository Index
83,087 visible repositories in the index
1–24 of 56 · Most stars
NumPy & SciPy for GPU
- cuda
- cudnn
- cublas
- cusolver
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Qiskit is an open-source SDK for working with quantum computers at the level of extended quantum circuits, operators, and primitives.
- quantum-computing
- qiskit
- sdk
- quantum-programming-language
🔩 Like builtins, but boltons. 250+ constructs, recipes, and snippets which extend (and rely on nothing but) the Python standard library. Nothing like Michael Bolton.
- python
- utilities
- cache
- data-structures
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
An open-source, GPU-accelerated physics simulation engine built upon NVIDIA Warp, specifically targeting roboticists and simulation researchers.
- physics-simulation
- robotics
- nvidia-warp
- newton-physics
TranslatedDeepAnalyze is the first agentic LLM for autonomous data science. 🎈Your AI data analyst, automatically analyzes large amounts of data and generates professional analysis reports with one click!
Mesa is an open-source Python library for agent-based modeling, ideal for simulating complex systems and exploring emergent behaviors.
- mesa
- agent-based-modeling
- agent-based-simulation
- simulation-framework
This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation.
POT : Python Optimal Transport
- optimal-transport
- numerical-optimization
- machine-learning
- emd
Single-cell analysis in Python. Scales to >100M cells.
- machine-learning
- data-science
- visualize-data
- transcriptomics
Reverse engineering SARS-CoV-2
- covid-19
Differentiable convex optimization layers
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
- differential-equations
- jax
- deep-learning
- neural-differential-equations
An implementation of chunked, compressed, N-dimensional arrays for Python.
- zarr
- ndimensional-arrays
- compressed
- python
Python Materials Genomics (pymatgen) is a robust materials analysis code that defines classes for structures and molecules with support for many electronic structure codes. It powers the Materials Project.
- materials
- science
- materials-science
- materials-informatics
A package for the sparse identification of nonlinear dynamical systems from data
- dynamical-systems
- machine-learning
- sparse-regression
- model-discovery
Elegant astronomy for Python
Fast and flexible physics-based battery models in Python
- pybamm
- battery-models
- solvers
- python
Deep probabilistic analysis of single-cell and spatial omics data
- scrna-seq
- variational-bayes
- variational-autoencoder
- cite-seq
Implementation of Alphafold 3 from Google Deepmind in Pytorch
- artificial-intelligence
- deep-learning
- attention-mechanisms
- denoising-diffusion
Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
- vizier
- hyperparameter-optimization
- tuning
- tuning-parameters
GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
- nvidia-warp
- mujoco-warp