Search Results - RepositoryStats

598

2.6k

mit

38

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

570 commits to master branch, last one about a month ago

get-started-with-JAX gordicaleksa

99

674

mit

9

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

24

212

apache-2.0

13

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

gpt jax nlp flax optax easydel transformers

Created 2023-04-09

1,914 commits to main branch, last one 9 days ago

flora-opt BorealisAI

5

84

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 5 months ago

diffusion-models-tutorial varun-ml

9

55

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

56 commits to master branch, last one about a month ago