chaoming0625's Stars
cupy/cupy
NumPy & SciPy for GPU
taichi-dev/taichi
Productive, portable, and performant GPU programming in Python.
stanford-crfm/levanter
Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
tensorflow/mlir-hlo
brian-team/brian2
Brian is a free, open source simulator for spiking neural networks.
rougier/scientific-posters
A collection of scientific posters (with sources) made with Pages or ComicLife
openxla/stablehlo
Backward compatible ML compute opset inspired by HLO/MHLO
iree-org/iree
A retargetable MLIR-based machine learning compiler and runtime toolkit.
openxla/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
tensor-compiler/taco
The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
hidet-org/hidet
An open-source efficient deep learning framework/compiler, written in python.
pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
google-research/tuning_playbook
A playbook for systematically maximizing the performance of deep learning models.
activeloopai/deeplake
Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangChain. Store, query, version, & visualize any AI data. Stream data in real-time to PyTorch/TensorFlow. https://activeloop.ai
LaPreprint/LaPreprint
📝 A nicely formatted LaTeX preprint template
jax-ml/jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
triton-lang/triton
Development repository for the Triton language and compiler
pytorch/xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
eriklindernoren/ML-From-Scratch
Machine Learning From Scratch. Bare bones NumPy implementations of machine learning models and algorithms with a focus on accessibility. Aims to cover everything from linear regression to deep learning.
c-xy17/NeuralModeling
for 神经计算建模实战
brainpy/BrainPy
Brain Dynamics Programming in Python
tensorflow/tensorflow
An Open Source Machine Learning Framework for Everyone
dfm/extending-jax
Extending JAX with custom C++ and CUDA code
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
valinet/ExplorerPatcher
This project aims to enhance the working environment on Windows
cloudhan/jax-windows-builder
A community supported Windows build for jax.
brainpy/examples
Examples of BrainPy
raphaelvallat/yasa
YASA (Yet Another Spindle Algorithm): a Python package to analyze polysomnographic sleep recordings.
numba/llvmlite
A lightweight LLVM python binding for writing JIT compilers
numba/numba
NumPy aware dynamic Python compiler using LLVM