JAX in JavaScript – ML library for the web, running on WebGPU & Wasm

High-Speed Snake Simulator

Electromagnetic FDTD Simulations in JAX

[WACV 2025] Official implementation of "RAW-Diffusion: RGB-Guided Diffusion Models for High-Fidelity RAW Image Generation"

Cell Tracking metrics and tools for the CellTrackingChallenge as described in 'CHOTA: A Higher Order Accuracy Metric for Cell Tracking'

A rich, visual interface for easily starting and monitoring your Hydra applications on SLURM clusters

Official implementation of "Learned Fourier Bases for Deep Set Feature Extractors in Automotive Reinforcement Learning"

A simple to set up 2D automated vehicle gym env featuring random problem generation and dynamic observations.

Official implementation of "Deep Reinforcement Learning for Autonomous Driving using High-Level Heterogeneous Graph Representations"

Clustering algorithms (Mean shift and K-Means) from scratch in NumPy, PyTorch, TensorFlow, and JAX

Official PyTorch implementation for ChimeraMix: Image Classification on Small Datasets via Masked Feature Mixing (IJCAI 2022)

Official implementation of Constrained Mean Shift Clustering

Build cross-platform Native Progressive Web Apps for iOS, Android, and the Web ⚡️