Gholamhossin Eslami's repositories
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
FlaxTrainer
Training module for Flax neural network library
genflax
Diffusion Models and Flax
graph-jepa
Graph-level Representation Learning with Joint-Embedding Predictive Architectures
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
JAXiFold
Differential Geometry in Jax
mayhem
The Python asyncio tutorial I wish existed earlier
neural-tangents
Fast and Easy Infinite Neural Networks in Python
pykan
Kolmogorov Arnold Networks
segment-anything-flax
The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.