A pure-functional implementation of a machine learning transformer model in Python/JAX

Development repository for the Triton language and compiler

CUDA on non-NVIDIA GPUs

MLX: An array framework for Apple silicon

A spinup acceleration procedure for land surface models (LSM) . Developer team Mandresy Rasolonjatovo, Tianzhang Cai, Matthew Archer, Daniel Goll

Stellarium is a free GPL software which renders realistic skies in real time with OpenGL. It is available for Linux/Unix, Windows and macOS. With Stellarium, you really see what you can see with yo…

A hyperparameter optimization framework

The simplest, fastest repository for training/finetuning medium-sized GPTs.

An evaluation framework for machine learning models simulating high-throughput materials discovery.

MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.

CUDA / C++ path tracer with BVH

Awesome resources on normalizing flows.

AlphaFold2 non-docker setup

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Morphological snakes for image segmentation and tracking

Das ist ein julia code. Central resource for the Cambridge julia study group

Performance benchmarks and regression tests for the ExCALIBUR project

Helpful PyTorch models and data-loading tools

Instant neural graphics primitives: lightning fast NeRF and more

Python package for automatic tree crown delineation based on the Detectron2 implementation of Mask R-CNN

Solve puzzles. Learn CUDA.

Tribuo - A Java machine learning library

Modified version of Alphafold to divide CPU part (MSA and template searching) and GPU part. This can accelerate Alphafold when predicting multiple structures

Compiler for multiple programming models (SYCL, C++ standard parallelism, HIP/CUDA) for CPUs and GPUs from all vendors: The independent, community-driven compiler for C++-based heterogeneous progra…

A benchmark framework for Tensorflow

Reference implementations of MLPerf™ training benchmarks