lukeconibear's Stars
google-research/google-research
Google Research
karpathy/nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
microsoft/Swin-Transformer
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
dabeaz-course/python-mastery
Advanced Python Mastery (course by @dabeaz)
XAMPPRocky/tokei
Count your code, quickly.
numba/numba
NumPy aware dynamic Python compiler using LLVM
google-research/vision_transformer
mwouts/jupytext
Jupyter Notebooks as Markdown Documents, Julia, Python or R scripts
facebookincubator/cinder
Cinder is Meta's internal performance-oriented production version of CPython.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
numba/llvmlite
A lightweight LLVM python binding for writing JIT compilers
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
keras-team/keras-core
A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
aesara-devs/aesara
Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.
198808xc/Pangu-Weather
An official implementation of Pangu-Weather
google/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
google/jax-cfd
Computational Fluid Dynamics in JAX
chengtan9907/OpenSTL
OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning
stanford-crfm/levanter
Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
amazon-science/earth-forecasting-transformer
Official implementation of Earthformer
ecmwf-lab/ai-models
dask/dask-kubernetes
Native Kubernetes integration for Dask
google/lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
stanford-crfm/haliax
Named Tensors for Legible Deep Learning in JAX
leap-stc/ClimSim
An open large-scale dataset for training high-resolution physics emulators in hybrid multi-scale climate simulators.
ecmwf-lab/ai-models-panguweather
seominseok0429/Implicit-Stacked-Autoregressive-Model-for-Video-Prediction
Implicit Stacked Autoregressive Model for Video Prediction (official implementation)
earthnet2021/earthnet-model-intercomparison-suite
ecmwf-lab/ai-models-fourcastnet
earthnet2021/earthnet-toolkit