Search Results - RepositoryStats

564

2.5k

mit

36

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

jax flax optax uvadlc pytorch tutorial tutorials deep-learning pytorch-lightning

Created 2020-07-25

567 commits to master branch, last one 3 months ago

get-started-with-JAX gordicaleksa

97

630

mit

8

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as t...

jax lax xla flax haiku numpy optax python jupyter tutorial learn-jax deep-learning machine-learning

Created 2021-10-28

33 commits to main branch, last one 2 years ago

EasyDeL erfanzar

23

191

apache-2.0

12

Accelerate, Optimize performance with streamlined training and serving options with JAX.

gpt jax nlp flax mojo optax easydel pytorch transformers machine-learning

Created 2023-04-09

1,742 commits to main branch, last one 10 hours ago

flora-opt BorealisAI

3

57

lgpl-3.0

3

This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.

jax flax lora peft optax transformers deep-learning random-projection large-language-models memory-efficient-tuning

Created 2024-02-05

6 commits to main branch, last one 3 months ago

diffusion-models-tutorial varun-ml

8

53

mit

2

Experiment with diffusion models that you can run on your local jupyter instances

jax vae optax pytorch dm-haiku diffusion-models

Created 2022-10-20

54 commits to master branch, last one about a year ago