Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
Projects in Awesome Lists tagged with optax
A curated list of projects in awesome lists tagged with optax .
https://github.com/phlippe/uvadlc_notebooks
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
deep-learning flax jax optax pytorch pytorch-lightning tutorial tutorials uvadlc
Last synced: 29 Nov 2024
https://github.com/gordicaleksa/get-started-with-jax
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 the content I found useful while learning about the JAX ecosystem.
deep-learning flax haiku jax jupyter lax learn-jax machine-learning numpy optax python tutorial xla
Last synced: 30 Dec 2024
https://github.com/erfanzar/easydel
Accelerate, Optimize performance with streamlined training and serving options with JAX.
easydel flax gpt jax nlp optax transformers
Last synced: 28 Dec 2024
https://github.com/erfanzar/EasyDeL
Accelerate, Optimize performance with streamlined training and serving options with JAX.
easydel flax gpt jax machine-learning mojo nlp optax transformers
Last synced: 16 Nov 2024
https://github.com/borealisai/flora-opt
This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.
deep-learning flax jax large-language-models lora memory-efficient-tuning optax peft random-projection transformers
Last synced: 29 Dec 2024
https://github.com/evanatyourservice/psgd_jax
Implementation of PSGD optimizer in JAX
jax optax optimization preconditioning psgd second-order-optimization
Last synced: 28 Nov 2024
https://github.com/frankroeder/goal_conditioned_rl
Goal-conditioned reinforcement learning like 🔥
actor-critic deep-deterministic-policy-gradient deep-reinforcement-learning distrax droq flax goal-conditioned-rl goals gymnasium gymnasium-robotics jax mpi4jax optax reinforcement-learning robotics soft-actor-critic
Last synced: 06 Dec 2024
https://github.com/jessefarebro/flax-mup
Maximal Update Parametrization (μP) with Flax & Optax.
Last synced: 30 Nov 2024
https://github.com/salfaris/vgae-jax
Variational Graph Autoencoder implemented using JAX+Jraph
deep-learning graph-neural-network haiku jax jraph machine-learning optax variational-autoencoder
Last synced: 15 Nov 2024
https://github.com/daniel-j-h/nedem
Neural implicit digital elevation model
dem digital-elevation-model flax glo-30 jax machine-learning optax
Last synced: 18 Dec 2024
https://github.com/activatedgeek/optax-swag
Stochastic Weight Averaging (SWA) transforms for Optax with JAX
jax optax python stochastic-weight-averaging swag
Last synced: 19 Dec 2024
https://github.com/bischtob/opterax
An gradient-free optimization suite written in JAX. We conform to the optax interface and provide ensemble-based optimizers.
jax machine-learning optax optimization
Last synced: 11 Dec 2024
https://github.com/evanatyourservice/sophia-schedulefree-jax
Sophia optimizer with ScheduleFree
Last synced: 28 Nov 2024
https://github.com/evanatyourservice/flat-sophia
sophia optimizer further projected towards flat areas of loss landscape
jax optax optimization second-order-optimization sophia
Last synced: 28 Nov 2024
https://github.com/alexjackson1/hk-mnist
An implementation of MNIST classification using LeNet-300-100 in JAX (using Haiku and Optax).
deep-learning haiku jax mnist optax
Last synced: 20 Dec 2024
https://github.com/shawonashraf/postagger-lstm-jax
LSTM POS Tagger implementation in Jax and Flax
Last synced: 15 Dec 2024
