https://github.com/tristandeleu/metax

A collection of extensions for meta-learning in JAX
https://github.com/tristandeleu/metax

jax meta-learning

Last synced: 20 days ago
JSON representation

A collection of extensions for meta-learning in JAX

• Host: GitHub
• URL: https://github.com/tristandeleu/metax
• Owner: tristandeleu
• License: mit
• Created: 2020-10-23T13:43:35.000Z (about 4 years ago)
• Default Branch: main
• Last Pushed: 2020-10-23T17:20:54.000Z (about 4 years ago)
• Last Synced: 2024-10-04T18:19:42.744Z (about 1 month ago)
• Topics: jax, meta-learning
• Language: Python
• Homepage:
• Size: 6.84 KB
• Stars: 8
• Watchers: 2
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Metax: Meta-learning extensions for [JAX](https://github.com/google/jax/)

[![PyPI](https://img.shields.io/pypi/v/metax)](https://pypi.org/project/metax/) [![Build Status](https://travis-ci.com/tristandeleu/metax.svg?branch=main)](https://travis-ci.com/tristandeleu/metax)

**This is an experimental library**

Metax is a collection of extensions for meta-learning in JAX. Metax contains a subset of the utility functions from [Torchmeta](https://github.com/tristandeleu/pytorch-meta).