SQA scottish Science Baccalaureate of Christopher Rae. Improving the preformance of distributed data parallelism of low wifi bandwidths

Common practices for distributed training using various backends

Simple Learning Pipeline using Jax

Code for various probabilistic deep learning models

Conversion of LAST library from JAX to PyTorch

A setup for solving shape optimization problems depending on a partial differential equation using boundary integral methods and the Python library JAX. The code was developed as part of a Bachelor thesis project in Engineering Mathematics at the Royal Institute of Technology in Spring 2024 by Rebecka Johansson and Asta Stensson.

Toy implementations of popular deep learning optimizers from scratch in JAX

This is an implementation of linear regression using JAX and a functional programming approach. You can find a related blog post at the following link: https://medium.com/@sahinadirhan/simple-linear-regression-using-jax-5ef2eefb8cf4

A JAX implementation of DEformer for arbitrary conditioning.

Machine learning samples

This tutorial is part of the "Intensive Course Machine Learning: From Basics to Advanced Concepts" of the QuCoLiMa research network and the International Max Planck Research School Physics of Light.

minimal forward-mode automatic differentiation using python's abstract syntax tree

A comprehensive programming environment designed to facilitate research and development at the intersection of high-energy physics and machine learning.

A generalized implementation of Grad-CAM for Flax

Python code for "Probabilistic Machine learning" book by Kevin Murphy

Searching for galaxy satellites by their impact on lens potential in strong gravitational lensing setup

Simple VAE example with Jax.

Vectorized Kalman filters in JAX

This is the official repository of the paper "RoCourseNet: Distributionally Robust Training of a Prediction Aware Recourse Model".