Model Card for Jat

This is a multi-modal and multi-task model.

Model Details

Model Description

• Developed by: The JAT Team
• License: Apache 2.0

Model Sources

• Repository: https://github.com/huggingface/jat
• Paper: https://huggingface.co/papers/2402.09844
• Demo: Coming soon

Training

The model was trained on the following tasks:

• Alien
• Amidar
• Assault
• Asterix
• Asteroids
• Atlantis
• Bank Heist
• Battle Zone
• Beam Rider
• Berzerk
• Bowling
• Boxing
• Breakout
• Centipede
• Chopper Command
• Crazy Climber
• Defender
• Demon Attack
• Double Dunk
• Enduro
• Fishing Derby
• Freeway
• Frostbite
• Gopher
• Gravitar
• H.E.R.O.
• Ice Hockey
• James Bond
• Kangaroo
• Krull
• Kung-Fu Master
• Montezuma's Revenge
• Ms. Pacman
• Name This Game
• Phoenix
• PitFall
• Pong
• Private Eye
• Q*Bert
• River Raid
• Road Runner
• Robotank
• Seaquest
• Skiing
• Solaris
• Space Invaders
• Star Gunner
• Surround
• Tennis
• Time Pilot
• Tutankham
• Up and Down
• Venture
• Video Pinball
• Wizard of Wor
• Yars Revenge
• Zaxxon
• Action Obj Door
• Blocked Unlock Pickup
• Boss Level No Unlock
• Boss Level
• Find Obj S5
• Go To Door
• Go To Imp Unlock
• Go To Local
• Go To Obj Door
• Go To Obj
• Go To Red Ball Grey
• Go To Red Ball No Dists
• Go To Red Ball
• Go To Red Blue Ball
• Go To Seq
• Go To
• Key Corridor
• Mini Boss Level
• Move Two Across S8N9
• One Room S8
• Open Door
• Open Doors Order N4
• Open Red Door
• Open Two Doors
• Open
• Pickup Above
• Pickup Dist
• Pickup Loc
• Pickup
• Put Next Local
• Put Next S7N4
• Synth Loc
• Synth Seq
• Synth
• Unblock Pickup
• Unlock Local
• Unlock Pickup
• Unlock To Unlock
• Unlock
• Assembly
• Basketball
• BinPicking
• Box Close
• Button Press Topdown Wall
• Button Press Topdown
• Button Press Wall
• Button Press
• Coffee Button
• Coffee Pull
• Coffee Push
• Dial Turn
• Disassemble
• Door Close
• Door Lock
• Door Open
• Door Unlock
• Drawer Close
• Drawer Open
• Faucet Close
• Faucet Open
• Hammer
• Hand Insert
• Handle Press Side
• Handle Press
• Handle Pull Side
• Handle Pull
• Lever Pull
• Peg Insert Side
• Peg Unplug Side
• Pick Out Of Hole
• Pick Place Wall
• Pick Place
• Plate Slide Back Side
• Plate Slide Back
• Plate Slide Side
• Plate Slide
• Push Back
• Push Wall
• Push
• Reach Wall
• Reach
• Shelf Place
• Soccer
• Stick Pull
• Stick Push
• Sweep Into
• Sweep
• Window Close
• Window Open
• Ant
• Inverted Double Pendulum
• Half Cheetah
• Hopper
• Humanoid
• Inverted Pendulum
• Pusher
• Reacher
• Humanoid Standup
• Swimmer
• Walker 2d

How to Get Started with the Model

Use the code below to get started with the model.

from transformers import AutoModelForCausalLM

model = AutoModelForCausalLM.from_pretrained("jat-project/jat")

Citation

@article{gallouedec2024jack,
    title = {{Jack of All Trades, Master of Some, a Multi-Purpose Transformer Agent}},
    author = {Gallouédec, Quentin and Beeching, Edward and Romac, Clément and Dellandréa, Emmanuel},
    journal = {arXiv preprint arXiv:2402.09844},
    year = {2024},
    url = {https://arxiv.org/abs/2402.09844}
}