feat(optim): add support of AdEMAMix optimizer #237

	name: pull_requests

	on:
	pull_request:
	branches: main

	jobs:
	docs:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v4
	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: 3.9
	architecture: x64
	- name: Cache python modules
	uses: actions/cache@v4
	with:
	path: ~/.cache/pip
	key: ${{ runner.os }}-python-${{ matrix.python }}-${{ hashFiles('pyproject.toml') }}-docs
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -e ".[docs]"

	- name: Build documentation
	run: cd docs && bash build.sh

	- name: Documentation sanity check
	run: test -e docs/build/index.html \|\| exit

	triage:
	permissions:
	contents: read
	pull-requests: write
	runs-on: ubuntu-latest
	steps:
	- uses: actions/labeler@v5
	with:
	repo-token: "${{ secrets.GITHUB_TOKEN }}"

Provide feedback